Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/python/
Dtest_util.py53 output_tensor_idx = subgraph.Outputs(output_idx)
54 output_tensor = subgraph.Tensors(output_tensor_idx)
/external/tensorflow/tensorflow/lite/delegates/
Dinterpreter_utils_test.cc115 const int output_tensor_idx = interpreter_->outputs()[0]; in TEST_P() local
122 TfLiteTensor* output_tensor = interpreter_->tensor(output_tensor_idx); in TEST_P()
Ddelegate_test_util.cc408 const int output_tensor_idx = interpreter_->outputs()[0]; in VerifyInvoke() local
413 TfLiteTensor* output_tensor = interpreter_->tensor(output_tensor_idx); in VerifyInvoke()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dobject_reader.cc144 int output_tensor_idx = node_->outputs->data[id]; in AddOutput() local
146 RETURN_IF_ERROR(ReadValueByTensorIdx(output_tensor_idx, &value)); in AddOutput()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.cc399 const int32_t output_tensor_idx = op->outputs.front(); in PassQuantizationAndGetConsumers() local
410 TensorT* output_tensor = subgraph->tensors[output_tensor_idx].get(); in PassQuantizationAndGetConsumers()
417 output_tensor_idx, output_tensor, in PassQuantizationAndGetConsumers()
418 GetTensorConsumers(model, subgraph, output_tensor_idx)); in PassQuantizationAndGetConsumers()
Dquantize_model_test.cc613 const int output_tensor_idx = 0; in TEST_F() local
621 subgraph->tensors[conv_op->outputs[output_tensor_idx]].get(); in TEST_F()
722 const int output_tensor_idx = 0; in TEST_P() local
730 subgraph->tensors[conv_op->outputs[output_tensor_idx]].get(); in TEST_P()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc112 const int output_tensor_idx = params->output_tensors->data[o]; in Create() local
115 if (delegate->static_unpacked_data_map_.count(output_tensor_idx) == 0) { in Create()
116 outputs.insert(output_tensor_idx); in Create()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc4120 int output_tensor_idx; in Invoke() local
4122 std::tie(output_tensor_idx, input_tensor_idx) = feedback_loop; in Invoke()
4123 TfLiteTensor& src = context->tensors[output_tensor_idx]; in Invoke()