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.cc46 int num_squeeze_dims = op_context.params->num_squeeze_dims; in Prepare() local
54 if (num_squeeze_dims == 0) { in Prepare()
62 for (int idx = 0; idx < num_squeeze_dims; ++idx) { in Prepare()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h338 int num_squeeze_dims; member
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc596 params->num_squeeze_dims = squeeze_dims->size(); in ParseOpData()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1655 Expect(builtin->num_squeeze_dims != 0, in Validate()
2409 builtin->num_squeeze_dims ? builtin->squeeze_dims : nullptr, in Map()
2410 static_cast<uint32_t>(builtin->num_squeeze_dims)); in Map()