Home
last modified time | relevance | path

Searched refs:pflr_ (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device_test.cc50 pflr_ = tensorflow::MakeUnique<ProcessFunctionLibraryRuntime>(
54 flr_ = pflr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0");
59 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr()
66 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::__anone5cc3cd90111::TestEnv
Dkernel_and_device.cc48 Status status = pflr_->ReleaseHandle(handle_); in ~KernelAndDeviceFunc()
88 function_def = pflr_->GetFLR(host_cpu_device_->name()) in Init()
146 pflr_->Instantiate(ndef.op(), AttrSlice(ndef), options, &handle_)); in Init()
147 return pflr_->GetOutputDevices(handle_, &output_devices_); in Init()
166 const std::vector<Device*> devices = pflr_->device_mgr()->ListDevices(); in Run()
316 Rendezvous* rendezvous = new IntraProcessRendezvous(pflr_->device_mgr()); in Run()
345 pflr_->Run(opts, handle_, input_vector, outputs, in Run()
Dcontext.h93 return pflr_->GetFLR(d->name()); in func_lib()
96 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr()
279 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
Dkernel_and_device.h201 pflr_(pflr), in KernelAndDeviceFunc()
231 ProcessFunctionLibraryRuntime* const pflr_; // non-null
Dcontext.cc62 pflr_(new ProcessFunctionLibraryRuntime( in EagerContext()
444 pflr_.reset(new ProcessFunctionLibraryRuntime( in StoreCollectiveOpsServer()
477 pflr_.reset(new ProcessFunctionLibraryRuntime( in InitializeRemote()
/external/tensorflow/tensorflow/compiler/jit/
Dcreate_xla_launch_op_test.cc74 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in Init()
77 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
83 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::CreateXlaLaunchOpTest
Dextract_outside_compilation_pass_test.cc248 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in ExtractOutsideCompilationTest()
251 auto flr = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in ExtractOutsideCompilationTest()
260 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::ExtractOutsideCompilationForFunctionTest
/external/tensorflow/tensorflow/core/common_runtime/data/
Dstandalone.cc93 op_params.function_library = pflr_->GetFLR("/device:CPU:0"); in MakeIterator()
117 pflr_(pflr), in Dataset()
121 MakeUnique<FunctionHandleCache>(pflr_->GetFLR("/device:CPU:0")); in Dataset()
Dstandalone.h112 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_threadpool_test.cc66 pflr_.reset(new ProcessFunctionLibraryRuntime( in Init()
69 flr0_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
195 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::__anonaa65071b0111::FunctionLibraryRuntimeTest
Dfunction_test.cc160 pflr_.reset(new ProcessFunctionLibraryRuntime( in Init()
163 flr0_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
164 flr1_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:1"); in Init()
165 flr2_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:2"); in Init()
366 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::__anona9a56aea0111::FunctionLibraryRuntimeTest
/external/tensorflow/tensorflow/core/kernels/data/
Dgraph_rewrite_dataset.h84 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_ = nullptr; variable
Ddataset_test_base.cc158 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in InitFunctionLibraryRuntime()
161 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in InitFunctionLibraryRuntime()
Ddataset_test_base.h187 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
Dmulti_device_iterator_ops.cc58 pflr_(std::move(pflr)), in MultiDeviceIterator()
358 const std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::data::__anon2db6daca0111::MultiDeviceIterator
Dgraph_rewrite_dataset.cc55 TF_RETURN_IF_ERROR(ctx->function_library()->Clone(&flib_def_, &pflr_, &lib_)); in Optimize()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.h463 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
Dxla_compiler.cc441 pflr_.reset(new ProcessFunctionLibraryRuntime( in XlaCompiler()
446 flib_runtime_ = pflr_->GetFLR(device_->name()); in XlaCompiler()