| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 2282 input_tensor, argument
2305 input_tensor, weight, bias, attr, scalars, algorithm argument
2315 def meta_mkl_linear(input_tensor, packed_weight, orig_weight, bias, batch_size): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_device_selection_test.cc | 899 auto input_tensor = context->tensors[node->inputs->data[0]]; in CustomNoOpNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | function_optimizer.cc | 1427 TensorId input_tensor = ParseTensorName(node.input(idx)); in RestoreTensorMapping() local
|
| H A D | arithmetic_optimizer.cc | 264 const TensorId input_tensor = ParseTensorName(consumer->input(i)); in UpdateConsumers() local
418 const string* input_tensor = input_tensors.front(); in AbsorbInputByOptimizedNodesGroup() local
|
| H A D | auto_mixed_precision_test.cc | 184 auto input_tensor = GenerateRandomTensorInRange<DT_FLOAT>( in TestSimpleUnaryInferOp() local
|
| H A D | remapper_test.cc | 2174 auto input_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | eager_service_impl_test.cc | 923 const tensorflow::Tensor* input_tensor = nullptr; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A D | flatbuffer_import.cc | 1246 auto input_tensor = func_inputs[i]; in ConvertSubgraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset_test_base.cc | 963 for (auto& input_tensor : input_tensors) { in MakeDatasetTensor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | mirrored_strategy_test.py | 120 def replica_fn(input_tensor): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | tracer.cpp | 347 auto input_tensor = input.toTensor(); in addInput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | function_test.py | 883 def LinearWithReuse(input_tensor, reuse=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_util.h | 949 const Tensor& input_tensor = context->input(idx_data_in); variable
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_vmap.py | 1021 def output_to_grad_is_vmapped(input_tensor): argument
1030 def input_to_grad_is_vmapped(input_tensor): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_device.cc | 891 void BaseGPUDevice::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice()
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_compiled_autograd.py | 832 def forward(self, input_tensor): argument
|
| H A D | test_mkldnn_pattern_matcher.py | 2473 def forward(self, input_tensor): argument
|
| H A D | test_cpu_repro.py | 224 def forward(self, input_tensor): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | native_ops.cpp | 625 auto& input_tensor = p_node->Input(i).toTensor(); in __anon75e5f0513b02() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device.cc | 703 TFE_TensorHandle* input_tensor = inputs[0]; in CopyToMesh() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_transformers.py | 1734 def pad_last_dim(input_tensor, alignment_size, slice: bool = False): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 660 ITensorProxyPtr input_tensor = params->inputs[0].tensor(); in TEST_F() local
701 ITensorProxyPtr input_tensor = converter_->network()->addInput( in TEST_F() local
9158 ITensorProxyPtr input_tensor = input.tensor(); in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_ops.py | 3799 def _swap_axis(input_tensor, dim_index, last_index, name=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | benchmarks_test.py | 1490 def _boolean_mask_fn(self, input_tensor, mask): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties.cc | 821 const TensorId input_tensor = ParseTensorName(function_node->input(i)); in UpdateFunction() local
|