| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| H A D | np_utils.py | 667 def reduce_all(input_tensor, axis=None, keepdims=False): argument
676 def reduce_any(input_tensor, axis=None, keepdims=False): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/optim/ |
| H A D | test_zero_redundancy_optimizer.py | 903 def closure_ddp(input_tensor=input_tensor): argument
909 def closure_sharded(input_tensor=input_tensor): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate.cc | 1233 auto& input_tensor = context_->tensors[node->inputs->data[0]]; in TransformPackIntoSupportedOps() local
1272 auto& input_tensor = context_->tensors[node->inputs->data[0]]; in TransformUnpackIntoSupportedOps() local
2409 const auto& input_tensor = context->tensors[node->inputs->data[0]]; in Validate() local
3366 const auto input_tensor = context->tensors[node->inputs->data[0]]; in Validate() local
3596 const TfLiteTensor& input_tensor = in Map() local
3649 const auto& input_tensor = context->tensors[input_id]; in Map() local
5281 auto input_tensor = context->tensors[sparse_weight_tid]; in DensifyAndDequantizeConstTensor() local
5689 auto& input_tensor = context->tensors[node->inputs->data[0]]; in AddOpsAndTensors() local
6105 auto& input_tensor = context->tensors[node->inputs->data[0]]; in AddOpsAndTensors() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/ |
| H A D | sparse_tensor.h | 496 inline Status SparseTensor::Split(const SparseTensor& input_tensor, in Split()
582 const SparseTensor& input_tensor, const gtl::ArraySlice<int64_t> start, in Slice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/testdata/ |
| H A D | double_op.py | 26 def double(input_tensor): argument
|
| H A D | double_op.cc | 37 const Tensor& input_tensor = context->input(0); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/calibrator/ |
| H A D | custom_aggregator_op.py | 26 def custom_aggregator(input_tensor, tensor_id: str): argument
|
| H A D | custom_aggregator_op.cc | 41 const Tensor& input_tensor = context->input(0); in Compute() local
|
| /aosp_15_r20/test/mlts/benchmark/jni/ |
| D | run_tflite.cpp | 258 auto* input_tensor = mTfliteInterpreter->tensor(input); in setInput() local
389 auto* input_tensor = mTfliteInterpreter->tensor(input); in benchmark() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/adding_an_op/ |
| H A D | cuda_op_kernel.cc | 39 const Tensor& input_tensor = context->input(0); in Compute() local
|
| H A D | zero_out_op_kernel_3.cc | 45 const Tensor& input_tensor = context->input(0); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/fuzzing/op_fuzzing/ |
| H A D | identity_fuzz.cc | 26 Tensor input_tensor(tensorflow::DT_INT8, in FuzzImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/ci_build/builds/user_ops/ |
| H A D | cuda_op_kernel.cc | 39 const Tensor& input_tensor = context->input(0); in Compute() local
|
| H A D | zero_out_op_kernel_1.cc | 42 const Tensor& input_tensor = context->input(0); in Compute() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/AIDL_V3/ |
| D | reverse.mod.py | 19 def test(name, axis_value, input_tensor, output_tensor, input_data, output_data): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | guarantee_const_op.cc | 32 const Tensor& input_tensor = ctx->input(0); in Compute() local
|
| H A D | string_strip_op.cc | 34 const Tensor* input_tensor; in Compute() local
|
| H A D | string_to_number_op.cc | 41 const Tensor* input_tensor; in Compute() local
|
| H A D | unicode_script_op.cc | 27 const Tensor* input_tensor; in Compute() local
|
| H A D | string_to_hash_bucket_fast_op.h | 37 const Tensor* input_tensor; in Compute() local
|
| /aosp_15_r20/external/federated-compute/fcp/aggregation/protocol/ |
| H A D | configuration.proto | 41 tensorflow.TensorSpecProto input_tensor = 2; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| H A D | string_split_fuzz.cc | 33 Tensor input_tensor(tensorflow::DT_STRING, TensorShape({})); in FuzzImpl() local
|
| H A D | string_split_v2_fuzz.cc | 33 Tensor input_tensor(tensorflow::DT_STRING, TensorShape({})); in FuzzImpl() local
|
| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/ |
| H A D | sum_test.py | 41 def forward(self, input_tensor, dim: int): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | op_def_library_pybind.cc | 435 const py::object& input_tensor, const ArgDef& input_arg, in GetInputType()
489 py::object input_tensor = GetInputTensor(input_name, keywords, op_def); in ExtractInputsAndAttrs() local
|