Home
last modified time | relevance | path

Searched refs:SetArgs (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_api_info_test.cc39 void SetArgs(const std::vector<ArgSpec>& input_args_spec, in SetArgs() function
57 SetArgs(input_args, output_args, sig); in PopulateFunction()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor_test.cc150 TF_ASSERT_OK(call_frame.SetArgs({V(1.0), V(1.0)})); in TEST_F()
178 TF_ASSERT_OK(call_frame.SetArgs({V(1.0)})); in TEST_F()
226 TF_ASSERT_OK(call_frame.SetArgs({V(1.0)})); in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dfunction_test.cc909 TF_EXPECT_OK(frame.SetArgs({})); in TEST()
911 HasError(frame.SetArgs({a}), "Invalid argument"); in TEST()
922 HasError(frame.SetArgs({}), "Invalid argument: Expects 2 arguments"); in TEST()
926 HasError(frame.SetArgs({a, c}), in TEST()
928 TF_EXPECT_OK(frame.SetArgs({a, b})); in TEST()
Dfunction.h281 Status SetArgs(gtl::ArraySlice<Tensor> args);
Dfunction.cc942 Status FunctionCallFrame::SetArgs(gtl::ArraySlice<Tensor> args) { in SetArgs() function in tensorflow::FunctionCallFrame
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc1010 s = frame->SetArgs(*remote_args); in RunRemote()
1090 s = frame->SetArgs(args); in Run()
Dfunction_test.cc106 TF_CHECK_OK(frame.SetArgs(args)); in Run()
302 TF_RETURN_IF_ERROR(frame.SetArgs(args)); in InstantiateAndRunViaCallFrameInterface()
Ddirect_session.cc775 const Status s = call_frame.SetArgs(feed_args); in Run()