Home
last modified time | relevance | path

Searched refs:input_tensor_idx (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator_test.cc79 int input_tensor_idx = interpreter->inputs()[i]; in TEST() local
80 TfLiteTensor* tensor = interpreter->tensor(input_tensor_idx); in TEST()
140 int input_tensor_idx = interpreter->inputs()[i]; in TEST() local
141 TfLiteTensor* tensor = interpreter->tensor(input_tensor_idx); in TEST()
213 int input_tensor_idx = interpreter->inputs()[i]; in TEST() local
214 TfLiteTensor* tensor = interpreter->tensor(input_tensor_idx); in TEST()
298 int input_tensor_idx = interpreter->inputs()[0]; in TEST() local
299 TfLiteTensor* tensor = interpreter->tensor(input_tensor_idx); in TEST()
358 int input_tensor_idx = interpreter->inputs()[0]; in TEST() local
359 TfLiteTensor* tensor = interpreter->tensor(input_tensor_idx); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/delegates/
Dinterpreter_utils_test.cc114 const int input_tensor_idx = interpreter_->inputs()[0]; in TEST_P() local
117 memcpy(interpreter_->typed_tensor<float>(input_tensor_idx), input.data(), in TEST_P()
Ddelegate_test_util.cc407 const int input_tensor_idx = interpreter_->inputs()[0]; in VerifyInvoke() local
410 memcpy(interpreter_->typed_tensor<float>(input_tensor_idx), input.data(), in VerifyInvoke()
/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibration_wrapper.cc194 int input_tensor_idx = interpreter_->inputs()[i]; in Prepare() local
195 if (interpreter_->ResizeInputTensor(input_tensor_idx, dims) != kTfLiteOk) { in Prepare()
226 int input_tensor_idx = interpreter_->inputs()[i]; in FeedTensor() local
227 if (!SetTensor(input_tensor_idx, input)) { in FeedTensor()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.cc406 const int32_t input_tensor_idx = op->inputs[input_idx.front()]; in PassQuantizationAndGetConsumers() local
409 const TensorT* input_tensor = subgraph->tensors[input_tensor_idx].get(); in PassQuantizationAndGetConsumers()
Dquantize_model_test.cc610 const int input_tensor_idx = 0; in TEST_F() local
617 subgraph->tensors[conv_op->inputs[input_tensor_idx]].get(); in TEST_F()
719 const int input_tensor_idx = 0; in TEST_P() local
726 subgraph->tensors[conv_op->inputs[input_tensor_idx]].get(); in TEST_P()
Dquantize_model.cc829 const int input_tensor_idx = op->inputs[property.inputs[0].first]; in QuantizeOpOutput() local
830 TensorT* input_tensor = subgraph->tensors[input_tensor_idx].get(); in QuantizeOpOutput()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc4121 int input_tensor_idx; in Invoke() local
4122 std::tie(output_tensor_idx, input_tensor_idx) = feedback_loop; in Invoke()
4124 TfLiteTensor& dest = context->tensors[input_tensor_idx]; in Invoke()