| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | c_api_test.cc | 63 TfLiteTensor* input_tensor = TfLiteInterpreterGetInputTensor(interpreter, 0); in TEST() local
124 TfLiteTensor* input_tensor = TfLiteInterpreterGetInputTensor(interpreter, 0); in TEST() local
342 TfLiteTensor* input_tensor = TfLiteInterpreterGetInputTensor(interpreter, 0); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops_test.cc | 104 Tensor input_tensor(data_type, input_shape); in TEST() local
138 Tensor input_tensor(data_type, input_shape); in CreateSplitTensorGraph() local
171 Tensor input_tensor(data_type, input_shape); in CreateSplitResourceGraph() local
799 Tensor input_tensor(data_type, input_shape); in CreateConcatTensorGraph() local
849 Tensor input_tensor(data_type, input_shape); in CreateConcatResourceGraph() local
1226 Tensor input_tensor(data_type, input_shape); in CreateRoundtripTensorGraph() local
1281 Tensor input_tensor(data_type, input_shape); in CreateRoundtripResourceGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| H A D | resnet50.py | 69 def call(self, input_tensor, training=False): argument
144 def call(self, input_tensor, training=False): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/ |
| H A D | import_tensorflow.cc | 347 tensorflow::Status ImportTensorData(const TensorProto& input_tensor, in ImportTensorData()
387 tensorflow::Status ImportFloatArray(const TensorProto& input_tensor, in ImportFloatArray()
405 tensorflow::Status ImportComplex64Array(const TensorProto& input_tensor, in ImportComplex64Array()
423 tensorflow::Status ImportQuint8Array(const TensorProto& input_tensor, in ImportQuint8Array()
440 tensorflow::Status ImportInt32Array(const TensorProto& input_tensor, in ImportInt32Array()
457 tensorflow::Status ImportUint32Array(const TensorProto& input_tensor, in ImportUint32Array()
474 tensorflow::Status ImportInt64Array(const TensorProto& input_tensor, in ImportInt64Array()
491 tensorflow::Status ImportBoolArray(const TensorProto& input_tensor, in ImportBoolArray()
518 tensorflow::Status ImportStringArray(const TensorProto& input_tensor, in ImportStringArray()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/ |
| H A D | feature_column_v2.py | 2484 def expand_dims(input_tensor): argument
2513 def _to_sparse_input_and_drop_ignore_values(input_tensor, ignore_value=None): argument
2634 def _transform_input_tensor(self, input_tensor): argument
2787 def _get_dense_tensor_for_input_tensor(self, input_tensor): argument
2819 def _get_sparse_tensors_for_input_tensor(self, input_tensor): argument
3395 def _transform_input_tensor(self, input_tensor): argument
3560 def _transform_input_tensor(self, input_tensor, state_manager=None): argument
3664 def _transform_input_tensor(self, input_tensor, state_manager=None): argument
3782 def _transform_input_tensor(self, input_tensor): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | fractional_avg_pool_op_test.py | 80 def _GetExpectedFractionalAvgPoolResult(self, input_tensor, row_seq, col_seq, argument
111 def _ValidateFractionalAvgPoolResult(self, input_tensor, pooling_ratio, argument
|
| H A D | fractional_max_pool_op_test.py | 80 def _GetExpectedFractionalMaxPoolResult(self, input_tensor, row_seq, col_seq, argument
111 def _ValidateFractionalMaxPoolResult(self, input_tensor, pooling_ratio, argument
|
| H A D | bias_op_base.py | 156 def bias_add(input_tensor, bias_tensor): argument
164 def bias_add_1(input_tensor): argument
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/mutators/ |
| H A D | InPlaceOperationMutator.cpp | 48 const auto input_tensor = input_edge->tensor(); in output_edges_are_separate_tensors() local
99 auto input_tensor = input_edge->tensor(); in try_in_place_depthwiseconv() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | quantize_model.cc | 62 const TensorT* input_tensor = subgraph->tensors[op->inputs[idx]].get(); in DuplicateBiasesWithMultipleUses() local
275 TfLiteStatus QuantizeBias(ModelT* model, const TensorT* input_tensor, in QuantizeBias()
725 TensorT* input_tensor = subgraph->tensors[op->inputs[input_idx]].get(); in ApplyConstraints() local
1109 TensorT* input_tensor = subgraph->tensors[input_tensor_idx].get(); in QuantizeOpOutput() local
1584 TensorT* input_tensor = in QuantizeBiases() local
1795 TensorT* input_tensor = in EnsureBiasScaleCompatibility() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/ |
| H A D | op_signature.cc | 187 const Tensor* input_tensor = in GetOpSignature() local
220 const Tensor* input_tensor = in GetOpSignature() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/ |
| H A D | benchmark_model.cc | 65 Tensor* input_tensor) { in InitializeTensor()
79 Tensor input_tensor(input.data_type, input.shape); in CreateTensorsFromInputInfo() local
130 for (const std::pair<string, tensorflow::Tensor>& input_tensor : in GetOutputShapes() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | multi_threaded_pg.py | 351 def allgather(self, output_tensors, input_tensor, opts=AllgatherOptions()): argument
357 def _allgather_base(self, output_tensor, input_tensor, opts=AllgatherOptions()): argument
385 def _reduce_scatter_base(self, output_tensor, input_tensor, opts=ReduceScatterOptions()): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_conv2d_acl.h | 31 arm_compute::Tensor input_tensor; member
42 arm_compute::Tensor input_tensor; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | pooling_ops_test.py | 28 def NHWCToNCHW(input_tensor): argument
43 def NCHWToNHWC(input_tensor): argument
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| H A D | nl_classifier.cc | 215 TfLiteTensor* input_tensor = FindTensorWithNameOrIndex( in Preprocess() local
327 auto input_tensor = FindTensorWithNameOrIndex( in Initialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| H A D | validator.cc | 186 TfLiteTensor* input_tensor = golden_validation_entrypoint->tensor( in CheckGoldenOutput() local
196 TfLiteTensor* input_tensor = in CheckGoldenOutput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | device_base.h | 90 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice()
273 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | string_split_op.cc | 175 const Tensor* input_tensor; in Compute() local
250 const Tensor* input_tensor; in Compute() local
|
| H A D | list_kernels.h | 597 const Tensor& input_tensor = c->input(0); in Compute() local
850 const Tensor& input_tensor = c->input(1); in Compute() local
918 const Tensor& input_tensor = c->input(0); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | mirror_pad.cc | 165 const TfLiteTensor* input_tensor; in Eval() local
264 const TfLiteTensor* input_tensor; in Prepare() local
|
| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_xnnpack_quantizer.py | 739 def forward(self, input_tensor, hidden_tensor): argument
800 def forward(self, input_tensor, hidden_tensor): argument
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/ |
| H A D | etnaviv_ml.c | 124 unsigned input_tensor = poperation->input_tensor->index; in lower_operations() local
151 unsigned input_tensor = poperations[0].input_tensor->index; in lower_operations() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | interpreter.h | 393 TfLiteTensor* input_tensor(size_t index) { return tensor(inputs()[index]); } in input_tensor() function
397 const TfLiteTensor* input_tensor(size_t index) const { in input_tensor() function
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/ |
| H A D | functional.py | 122 def _all_gather_base(output_tensor, input_tensor, group=group.WORLD): argument
349 def forward(ctx, output_tensor, input_tensor, group): argument
|