Home
last modified time | relevance | path

Searched refs:output_tensor_index (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_2d_gpu.h199 Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims);
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];
235 Index<3> output_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];
250 Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims);
253 input_tensor_index[0] = output_tensor_index[sp0];
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dgraph_info.cc79 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in Partition() local
80 tensor_epochs_[output_tensor_index] = kEpochNotReady; in Partition()
168 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in UpdateNode() local
169 tensor_epochs_[output_tensor_index] = current_epoch; in UpdateNode()
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/main/native/
H A Dnativesignaturerunner_jni.cc74 int output_tensor_index = GetOutputTensorIndex(output_name); in GetOutputIndex() local
75 if (output_tensor_index == -1) return -1; in GetOutputIndex()
79 if (output_tensor_index == tensor_idx) return count; in GetOutputIndex()
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc1981 int output_tensor_index = node.outputs->data[output_index]; in InitializeTensorReleaseMap() local
1982 TfLiteTensor* output_tensor = tensor(output_tensor_index); in InitializeTensorReleaseMap()
1984 tensor_to_last_op_index_[output_tensor_index] = node_index; in InitializeTensorReleaseMap()
2027 int output_tensor_index = node.outputs->data[output_index]; in MaybeReleaseDynamicTensors() local
2028 TfLiteTensor* output_tensor = tensor(output_tensor_index); in MaybeReleaseDynamicTensors()
2032 tensorIsInput(output_tensor_index) || in MaybeReleaseDynamicTensors()
2033 tensorIsOutput(output_tensor_index)) in MaybeReleaseDynamicTensors()
2035 auto it = tensor_to_last_op_index_.find(output_tensor_index); in MaybeReleaseDynamicTensors()
/aosp_15_r20/external/federated-compute/fcp/client/engine/
H A Dtflite_wrapper.cc192 for (int output_tensor_index : interpreter_->outputs()) { in ConstructOutputs() local
194 interpreter_->tensor(output_tensor_index)); in ConstructOutputs()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/
H A Dhexagon_delegate_kernel.cc182 int output_tensor_index = node->outputs->data[i]; in ResizeOutputTensors() local
183 TfLiteTensor* output_tensor = &context->tensors[output_tensor_index]; in ResizeOutputTensors()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dkernel.cc673 const auto output_tensor_index = node_data->outputs().TfLiteIndex(i); in ValidateOutputTensorShapeConsistency() local
674 TfLiteTensor* tfl_tensor = &context->tensors[output_tensor_index]; in ValidateOutputTensorShapeConsistency()
695 op_name, i, output_tensor_index, tfl_shape_string.c_str(), in ValidateOutputTensorShapeConsistency()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_import.cc1355 int output_tensor_index = op->outputs[pair.index()]; in ConvertSubgraph() local
1356 auto& tensor = *subgraph.tensors[output_tensor_index]; in ConvertSubgraph()
1359 vals_map[output_tensor_index] = stats_op->getResult(0); in ConvertSubgraph()
1361 vals_map[output_tensor_index] = pair.value(); in ConvertSubgraph()
/aosp_15_r20/external/tensorflow/tensorflow/lite/examples/ios/camera/
H A DCameraExampleViewController.mm344 const int output_tensor_index = interpreter->outputs()[0];
345 TfLiteTensor* output_tensor = interpreter->tensor(output_tensor_index);
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dxnnpack_delegate.cc4645 const int output_tensor_index = node->outputs->data[0]; in VisitTransposeConvNode() local
4646 const TfLiteTensor& output_tensor = tensors[output_tensor_index]; in VisitTransposeConvNode()
4649 output_tensor_index, node_index)); in VisitTransposeConvNode()
4651 output_tensor_index)); in VisitTransposeConvNode()
4653 logging_context, output_tensor, output_tensor_index, node_index)); in VisitTransposeConvNode()
4724 /*output_id=*/xnnpack_tensors[output_tensor_index], in VisitTransposeConvNode()