Searched refs:input_dims_size (Results 1 – 7 of 7) sorted by relevance
38 int input_dims_size = input_dims->size; in Prepare() local39 TF_LITE_ENSURE(context, input_dims_size >= 1); in Prepare()45 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(input_dims_size + 1); in Prepare()46 for (int i = 0; i < input_dims_size; i++) { in Prepare()51 output_shape->data[input_dims_size] = input_dims->data[input_dims_size - 1]; in Prepare()
39 int input_dims_size = input_dims->size; in Prepare() local40 TF_LITE_ENSURE(context, input_dims_size >= 2); in Prepare()46 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(input_dims_size); in Prepare()47 for (int i = 0; i < input_dims_size; i++) { in Prepare()
121 bool ValidIndex(const int* input_dims, const int input_dims_size, in ValidIndex() argument123 if (input_dims_size == 0) { in ValidIndex()128 for (int idx = input_dims_size - 1; idx >= 0; --idx) { in ValidIndex()143 int GetOffset(const int* input_dims, const int input_dims_size, in GetOffset() argument145 if (input_dims_size == 0) return 0; in GetOffset()149 for (int idx = 0; idx < input_dims_size; ++idx) { in GetOffset()
157 int32_t input_dims_size) { in TfLiteInterpreterResizeInputTensor() argument158 std::vector<int> dims{input_dims, input_dims + input_dims_size}; in TfLiteInterpreterResizeInputTensor()
176 int32_t input_dims_size);
296 const int input_dims_size = input1->dims->size; in FakeFusedRegistration() local297 TF_LITE_ENSURE(context, output->dims->size == input_dims_size); in FakeFusedRegistration()298 for (int i = 0; i < input_dims_size; ++i) { in FakeFusedRegistration()
238 int input_dims_size); in TfLiteInterpreterResizeInputTensor() argument