Home
last modified time | relevance | path

Searched refs:squeeze_dims (Results 1 – 5 of 5) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dsqueeze.cc50 const int* squeeze_dims = op_context.params->squeeze_dims; in Prepare() local
64 int current = squeeze_dims[idx] < 0 ? squeeze_dims[idx] + input_num_dims in Prepare()
65 : squeeze_dims[idx]; in Prepare()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/
A Dbuiltin_op_data.h390 int squeeze_dims[8]; member
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/core/api/
A Dflatbuffer_conversions.cc1936 const auto* squeeze_dims = schema_params->squeeze_dims(); in ParseSqueeze() local
1937 if (squeeze_dims != nullptr) { in ParseSqueeze()
1939 sizeof(params->squeeze_dims), squeeze_dims, params->squeeze_dims, in ParseSqueeze()
1941 params->num_squeeze_dims = squeeze_dims->size(); in ParseSqueeze()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/
A Dtypes.h1119 int32_t squeeze_dims[4]; member
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/schema/
A Dschema_generated.h6736 std::vector<int32_t> squeeze_dims;
6746 const flatbuffers::Vector<int32_t> *squeeze_dims() const {
6752 verifier.VerifyVector(squeeze_dims()) &&
6763 void add_squeeze_dims(flatbuffers::Offset<flatbuffers::Vector<int32_t>> squeeze_dims) {
6764 fbb_.AddOffset(SqueezeOptions::VT_SQUEEZE_DIMS, squeeze_dims);
6780 flatbuffers::Offset<flatbuffers::Vector<int32_t>> squeeze_dims = 0) {
6782 builder_.add_squeeze_dims(squeeze_dims);
6788 const std::vector<int32_t> *squeeze_dims = nullptr) {
6789 auto squeeze_dims__ = squeeze_dims ? _fbb.CreateVector<int32_t>(*squeeze_dims) : 0;
13273 … _e = squeeze_dims(); if (_e) { _o->squeeze_dims.resize(_e->size()); for (flatbuffers::uoffset_t _…
[all …]

Completed in 31 milliseconds