Home
last modified time | relevance | path

Searched refs:num_squeeze_dims (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dsqueeze.cc44 int num_squeeze_dims = op_context.params->num_squeeze_dims; in Prepare() local
52 if (num_squeeze_dims == 0) { in Prepare()
60 for (int idx = 0; idx < num_squeeze_dims; ++idx) { in Prepare()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h390 int num_squeeze_dims; member
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc653 params->num_squeeze_dims = squeeze_dims->size(); in ParseOpDataTfLite()
655 params->num_squeeze_dims = 0; in ParseOpDataTfLite()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1995 Expect(builtin->num_squeeze_dims != 0, in Validate()
2910 builtin->num_squeeze_dims ? builtin->squeeze_dims : nullptr, in Map()
2911 static_cast<uint32_t>(builtin->num_squeeze_dims)); in Map()