| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | ProcessGroupWrapper.cpp | 374 bool check_same_size(const std::vector<at::Tensor>& input_tensors) { in check_same_size()
|
| H A D | ProcessGroupNCCL.cpp | 2435 bool check_same_size(const std::vector<at::Tensor>& input_tensors) { in checkForNCCLErrorsInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | quantization_ops_test.cc | 391 std::vector<Tensor> input_tensors; in ConvertAndRun() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | inline_function_utils.cc | 125 std::vector<string> input_tensors(caller.in_edges().size()); in InputDevices() local
|
| H A D | direct_session_test.cc | 2667 std::vector<Tensor> input_tensors; in FeedFetchBenchmarkHelper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | tape.h | 1023 const string& op_type, const std::vector<TapeTensor>& input_tensors, in Accumulate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_experimental.cc | 605 TF_Tensor** input_tensors, in TFE_InferShapes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset_test_base.h | 527 std::vector<std::unique_ptr<Tensor>> input_tensors, in TestDataset()
|
| H A D | dataset_test_base.cc | 957 auto input_tensors = dataset_params.GetInputTensors(); in MakeDatasetTensor() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | jit_utils.py | 527 def checkTrace(self, func, reference_tensors, input_tensors=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| H A D | saved_model.cc | 756 const auto& input_tensors = multi_inputs[i]; in RunMultipleSignatures() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/client/ |
| H A D | tf_session_helper.cc | 213 std::vector<Tensor> input_tensors; in RunCallableHelper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties.cc | 1115 std::vector<const Tensor*> input_tensors(ic->num_inputs(), nullptr); in UpdateNode() local
1374 std::vector<const Tensor*> input_tensors(num_inputs, nullptr); in AddNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_concat_op.cc | 469 OpInputList input_tensors; in Compute() local
|
| /aosp_15_r20/external/executorch/extension/pybindings/ |
| H A D | pybindings.cpp | 679 std::vector<torch::executor::TensorImpl> input_tensors; in run_method() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | runtime_fallback_kernels.cc | 679 Argument<Chain> in_chain, RemainingArguments input_tensors, in RuntimeFallbackKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | common.h | 657 TfLiteIntArray* input_tensors; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_util.h | 706 OpInputList* input_tensors) { in GetMklInputList()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 530 OpConverterParams* params, const std::vector<ITensorProxyPtr> input_tensors, in ConcatenateTensors()
4128 std::vector<ITensorProxyPtr> input_tensors; in ConvertConcat() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | constant_folding.cc | 2328 std::vector<Tensor> input_tensors(3); in SimplifyStridedSlice() local
|
| H A D | arithmetic_optimizer.cc | 414 std::deque<const string*> input_tensors; in AbsorbInputByOptimizedNodesGroup() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | ops.py | 2187 def _control_flow_post_processing(self, input_tensors=None): argument
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_runtime.cc | 1811 std::vector<c10::IValue> input_tensors({inputs}); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate.cc | 6141 const TfLiteIntArray* input_tensors, const TfLiteIntArray* output_tensors, in BuildGraph()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 3614 self, output_tensor_lists, input_tensors, expected_tensors, group_id argument
|