| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | while.cc | 456 for (int tensor_index : body_subgraph->outputs()) { in Eval_dynamic() local
559 for (int tensor_index : body_subgraph->outputs()) { in Eval_static() local
|
| H A D | subgraph_test_util.cc | 76 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor()
619 int tensor_index, in CreateConstantInt32Tensor()
|
| H A D | if.cc | 178 for (int tensor_index : active_branch_subgraph.outputs()) { in Eval() local
|
| H A D | unidirectional_sequence_lstm_test_util.h | 273 int tensor_index, const std::vector<std::vector<int>>& input_shapes) { in AddLayerNormCoeffsTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/ |
| H A D | delegate.cc | 179 for (uint32_t tensor_index : input_refs) { in Prepare() local
188 for (uint32_t tensor_index : output_refs) { in Prepare() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| H A D | calibration_logger.h | 52 TfLiteStatus LogTensorValue(int subgraph_index, int tensor_index, in LogTensorValue()
|
| H A D | calibration_reader.cc | 45 int subgraph_index, tensor_index; in AddCalibrationToModel() local
|
| H A D | calibrator.cc | 280 for (auto tensor_index : tensor_indices) { in GetLoggableTensorIndices() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | save_restore_util.h | 32 int64_t tensor_index; member
|
| H A D | save_restore_util.cc | 140 int tensor_index = tensor_name_index_and_slice_specs.first; in BuildPerDeviceSave() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | interpreter_test.cc | 95 int tensor_index; in TEST() local
113 int tensor_index; in TEST() local
1331 for (int tensor_index = 0; tensor_index < 4; tensor_index++) { in SetUp() local
1524 for (int tensor_index = 0; tensor_index < num_tensors; tensor_index++) { in SetUp() local
|
| H A D | signature_runner.cc | 77 for (int tensor_index : subgraph_->outputs()) { in Invoke() local
|
| H A D | simple_planner_test.cc | 205 bool IsAllocated(int tensor_index) { in IsAllocated()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | tensor_array_ops.cc | 421 const Tensor* tensor_index; in Compute() local
504 const Tensor* tensor_index; in Compute() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | TensorIndexing.cpp | 16 std::ostream& operator<<(std::ostream& stream, const TensorIndex& tensor_index) { in operator <<()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | quantization_wrapper_utils.cc | 44 string CreateTensorName(int op_index, int tensor_index) { in CreateTensorName()
|
| H A D | model_utils.cc | 119 int tensor_index) { in HasBuffer()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ForeachUtils.h | 305 for (const auto& tensor_index : c10::irange(num_tensors)) { in _group_tensors_by_first_tensors_device_and_dtype() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/profiling/ |
| H A D | profile_summarizer.cc | 38 int tensor_index) { in GetTensorName()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/ |
| H A D | op_signature_test.cc | 112 TfLiteTensor* tensor(int tensor_index) { return &tensors_[tensor_index]; } in tensor()
|
| /aosp_15_r20/external/pytorch/torch/distributed/rpc/ |
| H A D | internal.py | 56 def _tensor_receiver(cls, tensor_index): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| H A D | call.cc | 267 for (int tensor_index : subgraph->outputs()) { in Eval() local
|
| H A D | call_test.cc | 41 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/model_modifier/ |
| H A D | validation_graph_builder.cc | 443 #define RET_CHECK_INDEX(tensor_index, buffer_index) \ in Buffers() argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | MultiTensorApply.cuh | 311 for (const auto& tensor_index : c10::irange(num_tensors)) { in multi_tensor_apply_for_fused_optimizer() local
|