Home
last modified time | relevance | path

Searched refs:input_tensor_index (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_2d_gpu.h201 Index<3> input_tensor_index;
202 input_tensor_index[0] = output_tensor_index[sp0];
203 input_tensor_index[1] = output_tensor_index[sp1];
204 input_tensor_index[2] = output_tensor_index[sp2];
206 int input_index = TensorIndexToFlat(input_tensor_index, input_dims);
237 Index<3> input_tensor_index;
238 input_tensor_index[0] = output_tensor_index[sp0];
239 input_tensor_index[1] = output_tensor_index[sp1];
240 input_tensor_index[2] = output_tensor_index[sp2];
242 int input_index_i = TensorIndexToFlat(input_tensor_index, input_dims);
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dgraph_info.cc141 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode() local
142 if (input_tensor_index != kTfLiteOptionalTensor && in UpdateNode()
143 tensor_epochs_[input_tensor_index] == kEpochNotReady) { in UpdateNode()
173 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode() local
174 if (input_tensor_index == kTfLiteOptionalTensor) { in UpdateNode()
177 int input_epoch = tensor_epochs_[input_tensor_index]; in UpdateNode()
180 current_subset.input_tensors.push_back(input_tensor_index); in UpdateNode()
186 input_subset.output_tensors.push_back(input_tensor_index); in UpdateNode()
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/main/native/
H A Dnativesignaturerunner_jni.cc61 int input_tensor_index = GetInputTensorIndex(input_name); in GetInputIndex() local
62 if (input_tensor_index == -1) return -1; in GetInputIndex()
66 if (input_tensor_index == tensor_idx) return count; in GetInputIndex()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
H A Dnl_classifier_c_api.cc42 .input_tensor_index = options->input_tensor_index, in NLClassifierFromFileAndOptions()
H A Dnl_classifier.cc217 options_.input_tensor_name, options_.input_tensor_index); in Preprocess()
329 options.input_tensor_name, options.input_tensor_index); in Initialize()
335 options.input_tensor_index), in Initialize()
H A Dnl_classifier_c_api.h48 int input_tensor_index; member
H A Dnl_classifier.h45 int input_tensor_index = 0; member
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/examples/task/text/desktop/
H A Dnl_classifier_demo.cc29 ABSL_FLAG(int, input_tensor_index, -1, "Input tensor index of the model.");
49 options.input_tensor_index = absl::GetFlag(FLAGS_input_tensor_index); in Classify()
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc1972 int input_tensor_index = node.inputs->data[input_index]; in InitializeTensorReleaseMap() local
1973 TfLiteTensor* input_tensor = tensor(input_tensor_index); in InitializeTensorReleaseMap()
1975 tensor_to_last_op_index_[input_tensor_index] = node_index; in InitializeTensorReleaseMap()
2008 int input_tensor_index = node.inputs->data[input_index]; in MaybeReleaseDynamicTensors() local
2009 TfLiteTensor* input_tensor = tensor(input_tensor_index); in MaybeReleaseDynamicTensors()
2013 tensorIsInput(input_tensor_index) || tensorIsOutput(input_tensor_index)) in MaybeReleaseDynamicTensors()
2015 auto it = tensor_to_last_op_index_.find(input_tensor_index); in MaybeReleaseDynamicTensors()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/
H A Dcoreml_delegate.mm62 int input_tensor_index = 0;
65 input_tensor_index = 2;
67 if (GetInput(context, node, input_tensor_index)->type != kTfLiteFloat32) {
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/
H A Dhexagon_delegate_kernel.cc169 int input_tensor_index = node->inputs->data[i]; in ResizeOutputTensors() local
170 TfLiteTensor* input_tensor = &context->tensors[input_tensor_index]; in ResizeOutputTensors()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dkernel.cc632 const auto input_tensor_index = node_data->inputs().TfLiteIndex(i); in ValidateOutputTensorShapeConsistency() local
633 TfLiteTensor* tfl_tensor = &context->tensors[input_tensor_index]; in ValidateOutputTensorShapeConsistency()
638 op_data_->shared_info.buffer_map->GetTensorPtr(input_tensor_index); in ValidateOutputTensorShapeConsistency()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/java/src/native/task/text/nlclassifier/
H A Dnl_classifier_jni.cc65 .input_tensor_index = env->CallIntMethod(java_nl_classifier_options, in ConvertJavaNLClassifierOptions()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/ios/task/text/nlclassifier/Sources/
H A DTFLNLClassifier.m45 .input_tensor_index = options.inputTensorIndex,
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_weights_test.cc454 uint32_t input_tensor_index = op->inputs()->Get(0); in TEST_F() local
456 quantized_graph->tensors()->Get(input_tensor_index); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_weights_test.cc523 uint32_t input_tensor_index = op->inputs()->Get(0); in TEST_F() local
525 quantized_graph->tensors()->Get(input_tensor_index); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dxnnpack_delegate.cc4615 const int input_tensor_index = node->inputs->data[2]; in VisitTransposeConvNode() local
4616 const TfLiteTensor& input_tensor = tensors[input_tensor_index]; in VisitTransposeConvNode()
4619 input_tensor_index, node_index)); in VisitTransposeConvNode()
4621 CheckTensorShape(logging_context, input_tensor, 4, input_tensor_index)); in VisitTransposeConvNode()
4623 logging_context, input_tensor, input_tensor_index, node_index)); in VisitTransposeConvNode()
4721 /*input_id=*/xnnpack_tensors[input_tensor_index], in VisitTransposeConvNode()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc5229 int input_tensor_index = -1; in AddDequantizeOperatorsWhereNeeded() local
5235 input_tensor_index = 0; in AddDequantizeOperatorsWhereNeeded()
5242 input_tensor_index = 0; in AddDequantizeOperatorsWhereNeeded()
5252 int tensor_id = node->inputs->data[input_tensor_index]; in AddDequantizeOperatorsWhereNeeded()