Searched refs:proc_flr_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 139 proc_flr_.reset(new ProcessFunctionLibraryRuntime( in Init()
164 proc_flr_->AddCompositeDevice(d); in AddCompositeDevice()
171 return proc_flr_->Instantiate(name, attrs, instantiate_opts, handle); in Instantiate()
276 name, opts, attrs, instantiate_opts, args, rets, proc_flr_.get()); in Run()
286 name, opts, attrs, instantiate_opts, args, rets, proc_flr_.get()); in RunWithPackedArgs()
304 proc_flr_->Run(opts, handle, args, &out, [&status, &done](const Status& s) { in RunInstantiated()
328 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr_; member in tensorflow::__anon96a4e55d0111::ProcessFunctionLibraryRuntimeTest
351 proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:0"); in TEST_F()
354 flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0"); in TEST_F()
357 flr = proc_flr_->GetFLR("/device:CPU:0"); in TEST_F()
[all …]
|