Searched refs:tensor_args (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/autograph/utils/ |
| D | py_func.py | 67 tensor_args = []
76 tensor_args_idx[i] = len(tensor_args)
77 tensor_args.append(args[i])
96 tensor_args_idx[k] = len(tensor_args)
97 tensor_args.append(kwargs[k])
120 def f_wrapper(*tensor_args): argument
121 f_args = tuple(tensor_args[tensor_args_idx[i]] if arg_is_tensor[i] else a
124 k: tensor_args[tensor_args_idx[k]] if kwarg_is_tensor[k] else kwargs[k]
132 return script_ops.eager_py_func(f_wrapper, tensor_args, return_dtypes)
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 68 explicit EagerKernelArgs(gtl::InlinedVector<TensorValue, 4>&& tensor_args) in EagerKernelArgs() argument
69 : tensor_args_(std::move(tensor_args)) {} in EagerKernelArgs()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 714 gtl::InlinedVector<TensorValue, 4>&& tensor_args, in TestExecuteNodeArgs() argument
717 : EagerKernelArgs(std::move(tensor_args)), in TestExecuteNodeArgs()
873 gtl::InlinedVector<TensorValue, 4> tensor_args = {TensorValue()}; in TEST_F() local
875 std::move(tensor_args), in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 817 gtl::InlinedVector<TensorValue, 4>&& tensor_args) { in TestFunctionPackedArgs() argument
818 packed_args_.emplace(index, std::move(tensor_args)); in TestFunctionPackedArgs()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 1235 tensor_args = []
1239 tensor_args.append(arg)
1248 return script_ops.py_func(inner_f, tensor_args, [])
|