Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator_test.cc78 int input_tensor_idx = interpreter->inputs()[i]; in TEST() local
79 TfLiteTensor* tensor = interpreter->tensor(input_tensor_idx); in TEST()
139 int input_tensor_idx = interpreter->inputs()[i]; in TEST() local
140 TfLiteTensor* tensor = interpreter->tensor(input_tensor_idx); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.cc235 auto input_tensor_idx = op->inputs[op_tensor_info->activation_input_index]; in QuantizeOpWithBias() local
236 if (IsSubgraphInput(input_tensor_idx)) { in QuantizeOpWithBias()
237 TF_LITE_ENSURE_STATUS(AsymmetricQuantizeTensor(op_code, input_tensor_idx)); in QuantizeOpWithBias()
251 const TensorT* input_tensor = subgraph_->tensors[input_tensor_idx].get(); in QuantizeOpWithBias()
Dsubgraph_quantizer_test.cc78 const int input_tensor_idx = 0; in TEST() local
85 subgraph->tensors[conv_op->inputs[input_tensor_idx]].get(); in TEST()
177 const int input_tensor_idx = 0; in TEST() local
184 subgraph->tensors[conv_op->inputs[input_tensor_idx]].get(); in TEST()
/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibration_wrapper.cc103 int input_tensor_idx = interpreter_->inputs()[i]; in FeedTensor() local
104 if (!SetTensor(input_tensor_idx, input)) { in FeedTensor()