Searched defs:tensor_args (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/utils/ |
| H A D | py_func.py | 116 def f_wrapper(*tensor_args):
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | CPUFallback.cpp | 100 std::vector<at::Tensor> tensor_args; in cpu_fallback() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_eager_fallback.cpp | 216 std::vector<at::Tensor> tensor_args; in ts_eager_fallback() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | kernel_and_device.h | 74 explicit EagerKernelArgs(gtl::InlinedVector<TensorValue, 4>&& tensor_args) in EagerKernelArgs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | eager_service_impl_test.cc | 720 gtl::InlinedVector<TensorValue, 4>&& tensor_args, in TestExecuteNodeArgs()
879 gtl::InlinedVector<TensorValue, 4> tensor_args = {TensorValue()}; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | argument_spec.h | 162 std::vector<ArgumentInfo> tensor_args; member
|
| /aosp_15_r20/external/pytorch/torch/csrc/dynamo/ |
| H A D | compiled_autograd.h | 200 TensorArgs tensor_args; member
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 5220 tensor_args, argument
5445 def find_device(tensor_args, example_output): argument
5716 tensor_args, argument
7864 tensor_args, argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | builtin.py | 662 def tensor_args(self, *args): member in BuiltinVariable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | process_function_library_runtime_test.cc | 866 gtl::InlinedVector<TensorValue, 4>&& tensor_args) { in TestFunctionPackedArgs()
|