/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | kernel_and_device_test.cc | 50 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
|
D | kernel_and_device.cc | 48 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()
|
D | context.h | 93 return pflr_->GetFLR(d->name()); in func_lib() 96 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr() 279 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
D | kernel_and_device.h | 201 pflr_(pflr), in KernelAndDeviceFunc() 231 ProcessFunctionLibraryRuntime* const pflr_; // non-null
|
D | context.cc | 62 pflr_(new ProcessFunctionLibraryRuntime( in EagerContext() 444 pflr_.reset(new ProcessFunctionLibraryRuntime( in StoreCollectiveOpsServer() 477 pflr_.reset(new ProcessFunctionLibraryRuntime( in InitializeRemote()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | create_xla_launch_op_test.cc | 74 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
|
D | extract_outside_compilation_pass_test.cc | 248 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/ |
D | standalone.cc | 93 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()
|
D | standalone.h | 112 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | function_threadpool_test.cc | 66 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
|
D | function_test.cc | 160 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/ |
D | graph_rewrite_dataset.h | 84 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_ = nullptr; variable
|
D | dataset_test_base.cc | 158 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in InitFunctionLibraryRuntime() 161 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in InitFunctionLibraryRuntime()
|
D | dataset_test_base.h | 187 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
D | multi_device_iterator_ops.cc | 58 pflr_(std::move(pflr)), in MultiDeviceIterator() 358 const std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::data::__anon2db6daca0111::MultiDeviceIterator
|
D | graph_rewrite_dataset.cc | 55 TF_RETURN_IF_ERROR(ctx->function_library()->Clone(&flib_def_, &pflr_, &lib_)); in Optimize()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_compiler.h | 463 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
D | xla_compiler.cc | 441 pflr_.reset(new ProcessFunctionLibraryRuntime( in XlaCompiler() 446 flib_runtime_ = pflr_->GetFLR(device_->name()); in XlaCompiler()
|