Searched refs:output_tensor_index (Results 1 – 10 of 10) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | conv_2d_gpu.h | 199 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 D | graph_info.cc | 79 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 D | nativesignaturerunner_jni.cc | 74 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 D | subgraph.cc | 1981 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 D | tflite_wrapper.cc | 192 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 D | hexagon_delegate_kernel.cc | 182 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 D | kernel.cc | 673 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 D | flatbuffer_import.cc | 1355 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 D | CameraExampleViewController.mm | 344 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 D | xnnpack_delegate.cc | 4645 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()
|