Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/python/
Dtest_util.py49 output_tensor_idx = subgraph.Outputs(output_idx)
50 output_tensor = subgraph.Tensors(output_tensor_idx)
Dutil.py916 output_tensor_idx = op.outputs[0]
917 if output_tensor_idx in redundant_quant_tensors:
918 requantize_op = redundant_quant_tensors[output_tensor_idx]
930 output_tensor_idx = op.outputs[0]
931 if output_tensor_idx in output_dequant_tensors:
932 dequant_op = output_dequant_tensors[output_tensor_idx]
Dconvert.py917 for output_tensor_idx in op.outputs:
918 read_only_input_tensor_indices.discard(output_tensor_idx)
928 for output_tensor_idx in subgraph.outputs:
929 read_only_input_tensor_indices.discard(output_tensor_idx)
/external/tensorflow/tensorflow/lite/delegates/
Dinterpreter_utils_test.cc116 const int output_tensor_idx = interpreter_->outputs()[0]; in TEST_P() local
123 TfLiteTensor* output_tensor = interpreter_->tensor(output_tensor_idx); in TEST_P()
Ddelegate_test_util.cc459 const int output_tensor_idx = interpreter_->outputs()[0]; in VerifyInvoke() local
464 TfLiteTensor* output_tensor = interpreter_->tensor(output_tensor_idx); in VerifyInvoke()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dobject_reader.cc157 int output_tensor_idx = node_->outputs->data[id]; in AddOutput() local
159 RETURN_IF_ERROR(ReadValueByTensorIdx(output_tensor_idx, &value)); in AddOutput()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights_portable.cc354 const int32_t output_tensor_idx = op->outputs.front(); in PassQuantizationAndGetConsumers() local
365 TensorT* output_tensor = subgraph->tensors[output_tensor_idx].get(); in PassQuantizationAndGetConsumers()
372 output_tensor_idx, output_tensor, in PassQuantizationAndGetConsumers()
373 GetTensorConsumers(model, subgraph, output_tensor_idx)); in PassQuantizationAndGetConsumers()
Dquantize_weights.cc368 const int32_t output_tensor_idx = op->outputs.front(); in PassQuantizationAndGetConsumers() local
379 TensorT* output_tensor = subgraph->tensors[output_tensor_idx].get(); in PassQuantizationAndGetConsumers()
386 output_tensor_idx, output_tensor, in PassQuantizationAndGetConsumers()
387 GetTensorConsumers(model, subgraph, output_tensor_idx)); in PassQuantizationAndGetConsumers()
Dquantize_model_test.cc628 const int output_tensor_idx = 0; in TEST_F() local
636 subgraph->tensors[conv_op->outputs[output_tensor_idx]].get(); in TEST_F()
739 const int output_tensor_idx = 0; in TEST_P() local
747 subgraph->tensors[conv_op->outputs[output_tensor_idx]].get(); in TEST_P()
843 const int output_tensor_idx = 0; in TEST_P() local
851 subgraph->tensors[conv_op->outputs[output_tensor_idx]].get(); in TEST_P()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc977 for (const int output_tensor_idx : outputs_) { in ReleaseMemory() local
978 if (output_tensor_idx == kTfLiteOptionalTensor) continue; in ReleaseMemory()
979 TfLiteTensor* output_tensor = tensor(output_tensor_idx); in ReleaseMemory()
1170 const int output_tensor_idx = node.outputs->data[i]; in PrepareOpsAndTensors() local
1171 if (output_tensor_idx == kTfLiteOptionalTensor) continue; in PrepareOpsAndTensors()
1173 context(), custom_allocations_, output_tensor_idx)); in PrepareOpsAndTensors()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
Dquantize_model_test.cc432 const int output_tensor_idx = 0; in TEST_P() local
440 subgraph->tensors[conv_op->outputs[output_tensor_idx]].get(); in TEST_P()
537 const int output_tensor_idx = 0; in TEST_P() local
545 subgraph->tensors[conv_op->outputs[output_tensor_idx]].get(); in TEST_P()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc263 const int output_tensor_idx = params->output_tensors->data[o]; in Create() local
266 if (delegate.static_unpacked_data_map_.count(output_tensor_idx) == 0) { in Create()
267 outputs.insert(output_tensor_idx); in Create()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc5207 int output_tensor_idx; in Invoke() local
5209 std::tie(output_tensor_idx, input_tensor_idx) = feedback_loop; in Invoke()
5210 TfLiteTensor& src = context->tensors[output_tensor_idx]; in Invoke()