Home
last modified time | relevance | path

Searched refs:GetRemoteArg (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node_test.cc125 TF_EXPECT_OK(args.GetRemoteArg(FunctionArgIndex(1, 1), &remote0)); in TEST()
129 TF_EXPECT_OK(args.GetRemoteArg(FunctionArgIndex(2), &remote1)); in TEST()
Dexecute_node.h65 Status GetRemoteArg(const FunctionArgIndex& index, in GetRemoteArg() function
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.h52 virtual Status GetRemoteArg(const FunctionArgIndex& index, in GetRemoteArg() function
Dprocess_function_library_runtime.cc1688 TF_RETURN_IF_ERROR(args.GetRemoteArg(index, &remote_handle)); in Run()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc722 Status GetRemoteArg(const FunctionArgIndex& index, in GetRemoteArg() function in tensorflow::eager::__anon58c807940111::FunctionWithRemoteInputsTest::TestExecuteNodeArgs