Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/
Dgraph_info.cc131 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode() local
132 if (input_tensor_index != kTfLiteOptionalTensor && in UpdateNode()
133 tensor_epochs_[input_tensor_index] == kEpochNotReady) { in UpdateNode()
158 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode() local
159 if (input_tensor_index == kTfLiteOptionalTensor) { in UpdateNode()
162 int input_epoch = tensor_epochs_[input_tensor_index]; in UpdateNode()
165 current_subset.input_tensors.push_back(input_tensor_index); in UpdateNode()
171 input_subset.output_tensors.push_back(input_tensor_index); in UpdateNode()
/external/tensorflow/tensorflow/core/kernels/
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 …]
/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
Dnl_classifier_c_api.cc42 .input_tensor_index = options->input_tensor_index, in NLClassifierFromFileAndOptions()
Dnl_classifier_c_api.h48 int input_tensor_index; member
Dnl_classifier.cc191 options_.input_tensor_name, options_.input_tensor_index); in Preprocess()
303 options.input_tensor_name, options.input_tensor_index); in Initialize()
309 options.input_tensor_index), in Initialize()
Dnl_classifier.h45 int input_tensor_index = 0; member
/external/tflite-support/tensorflow_lite_support/examples/task/text/desktop/
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()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc167 int input_tensor_index = node->inputs->data[i]; in ResizeOutputTensors() local
168 TfLiteTensor* input_tensor = &context->tensors[input_tensor_index]; in ResizeOutputTensors()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc531 const auto input_tensor_index = node_data->inputs().TfLiteIndex(i); in ValidateOutputTensorShapeConsistency() local
532 TfLiteTensor* tfl_tensor = &context->tensors[input_tensor_index]; in ValidateOutputTensorShapeConsistency()
537 op_data_->buffer_map->GetTensorPtr(input_tensor_index); in ValidateOutputTensorShapeConsistency()
/external/tensorflow/tensorflow/lite/delegates/coreml/
Dcoreml_delegate.mm62 int input_tensor_index = 0;
65 input_tensor_index = 2;
67 if (GetInput(context, node, input_tensor_index)->type != kTfLiteFloat32) {
/external/tflite-support/tensorflow_lite_support/java/src/native/task/text/nlclassifier/
Dnl_classifier_jni.cc65 .input_tensor_index = env->CallIntMethod(java_nl_classifier_options, in ConvertJavaNLClassifierOptions()
/external/tflite-support/tensorflow_lite_support/ios/task/text/nlclassifier/Sources/
DTFLNLClassifier.m45 .input_tensor_index = options.inputTensorIndex,
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights_test.cc446 uint32_t input_tensor_index = op->inputs()->Get(0); in TEST_F() local
448 quantized_graph->tensors()->Get(input_tensor_index); in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc4142 int input_tensor_index = -1; in AddDequantizeOperatorsWhereNeeded() local
4148 input_tensor_index = 0; in AddDequantizeOperatorsWhereNeeded()
4155 input_tensor_index = 0; in AddDequantizeOperatorsWhereNeeded()
4165 int tensor_id = node->inputs->data[input_tensor_index]; in AddDequantizeOperatorsWhereNeeded()