Searched refs:GetLocalArg (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node.cc | 124 Status ExecuteNodeArgs::GetLocalArg(const FunctionArgIndex& index, in GetLocalArg() function in tensorflow::ExecuteNodeArgs
126 Status s = EagerKernelArgs::GetLocalArg(index, val); in GetLocalArg()
|
| D | execute_node_test.cc | 117 TF_EXPECT_OK(args.GetLocalArg(FunctionArgIndex(0), &local0)); in TEST()
121 TF_EXPECT_OK(args.GetLocalArg(FunctionArgIndex(1, 0), &local1)); in TEST()
|
| D | execute_node.h | 58 Status GetLocalArg(const FunctionArgIndex& index, Tensor* val) const override;
|
| D | kernel_and_device.h | 76 Status GetLocalArg(const FunctionArgIndex& index, Tensor* val) const override;
|
| D | kernel_and_device.cc | 53 Status EagerKernelArgs::GetLocalArg(const FunctionArgIndex& index, in GetLocalArg() function in tensorflow::EagerKernelArgs
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 46 virtual Status GetLocalArg(const FunctionArgIndex& index,
|
| D | process_function_library_runtime_test.cc | 825 Status GetLocalArg(const FunctionArgIndex& index, in GetLocalArg() function in tensorflow::__anon96a4e55d0111::TestFunctionPackedArgs
|
| D | process_function_library_runtime.cc | 1684 if (args.GetLocalArg(index, &tensor).ok()) { in Run()
|