Home
last modified time | relevance | path

Searched refs:dim_array (Results 1 – 3 of 3) sorted by relevance

/external/tflite-support/tensorflow_lite_support/custom_ops/
Dtflite_inference_main.cc32 const TfLiteIntArray* dim_array, in FillRandomString() argument
35 for (size_t i = 0; i < dim_array->size; i++) { in FillRandomString()
36 num_elements *= dim_array->data[i]; in FillRandomString()
/external/tensorflow/tensorflow/lite/tools/benchmark/
Dbenchmark_tflite_model.cc90 int GetNumElements(const TfLiteIntArray* dim_array) { in GetNumElements() argument
92 for (size_t i = 0; i < dim_array->size; i++) { in GetNumElements()
93 num_elements *= dim_array->data[i]; in GetNumElements()
99 const TfLiteIntArray* dim_array, in FillRandomString() argument
101 int num_elements = GetNumElements(dim_array); in FillRandomString()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1143 TfLiteIntArray* dim_array = TfLiteIntArrayCreate(dims.size()); in AddNewInputConstantTensor() local
1144 dim_array->size = dims.size(); in AddNewInputConstantTensor()
1145 std::copy(dims.begin(), dims.end(), dim_array->data); in AddNewInputConstantTensor()
1148 nn_type, type, dim_array, tensor_value, quant_params, tensor_index); in AddNewInputConstantTensor()
1149 TfLiteIntArrayFree(dim_array); in AddNewInputConstantTensor()