Home
last modified time | relevance | path

Searched refs:num_squeeze_dims (Results 1 – 6 of 6) 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.h303 int num_squeeze_dims; member
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc667 builtin->num_squeeze_dims ? builtin->squeeze_dims : nullptr, in Map()
668 static_cast<uint32_t>(builtin->num_squeeze_dims)); in Map()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc533 params->num_squeeze_dims = squeeze_dims->Length(); in ParseOpData()
/external/tensorflow/tensorflow/contrib/lite/
Dnnapi_delegate.cc430 static_cast<uint32_t>(builtin->num_squeeze_dims)); in AddOpsAndParams()
/external/tensorflow/tensorflow/lite/
Dnnapi_delegate.cc400 static_cast<uint32_t>(builtin->num_squeeze_dims)); in AddOpsAndParams()