| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 51 pflr_ = tensorflow::MakeUnique<ProcessFunctionLibraryRuntime>(
56 flr_ = pflr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0");
61 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr()
68 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::__anon781c7a9b0111::TestEnv
|
| D | kernel_and_device.cc | 84 Status status = pflr_->ReleaseHandle(handle_); in ~KernelAndDeviceFunc()
126 function_def = pflr_->GetFLR(host_cpu_device_->name()) in InstantiateFunc()
196 pflr_->Instantiate(ndef.op(), AttrSlice(ndef), options, &handle_)); in InstantiateFunc()
197 return pflr_->IsCrossProcess(handle_, &is_cross_process_); in InstantiateFunc()
203 return pflr_->GetOutputDevices(handle_, &output_devices_); in Init()
364 pflr_->Run(*opts, handle_, inputs, outputs, in Run()
|
| D | kernel_and_device.h | 243 pflr_(pflr), in KernelAndDeviceFunc()
254 pflr_->device_mgr()->ListDevices(); in KernelAndDeviceFunc()
293 ProcessFunctionLibraryRuntime* const pflr_; // non-null
|
| D | context.cc | 134 pflr_.reset(new eager::EagerProcessFunctionLibraryRuntime( in ResetPFLR()
138 pflr_.reset(new ProcessFunctionLibraryRuntime( in ResetPFLR()
811 const ConfigProto* config = pflr_ ? pflr_->config() : nullptr; in StoreCollectiveOpsServer()
997 const auto* config = pflr_->config(); in UpdateRemoteMaster()
1066 const auto* config = pflr_->config(); in SetMasterContextState()
1168 const auto* config = pflr_->config(); in InitializeRemoteWorker()
1223 const auto* config = pflr_->config(); in UpdateRemoteWorker()
|
| D | context.h | 134 return pflr_->GetFLR(d->name()); in func_lib()
137 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr()
509 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator_test.cc | 75 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in Init()
79 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
85 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::XlaKernelCreatorTest
|
| D | compilability_check_util_test.cc | 84 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in GetFunctionLibraryRuntime()
88 return pflr_->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in GetFunctionLibraryRuntime()
99 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::__anon1b55dcbf0111::CompilabilityCheckUtilTest
|
| D | extract_outside_compilation_pass_test.cc | 248 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in ExtractOutsideCompilationTest()
252 auto flr = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in ExtractOutsideCompilationTest()
261 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::ExtractOutsideCompilationForFunctionTest
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 81 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in OpsTestBase()
105 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in SetDevice()
167 params_->function_library = pflr_->GetFLR(device_->name()); in RunOpKernel()
|
| D | ops_testutil.h | 185 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()
194 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::__anon6d08b3dd0111::FunctionLibraryRuntimeTest
|
| D | function_test.cc | 166 pflr_.reset(new ProcessFunctionLibraryRuntime( in Init()
169 flr0_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
170 flr1_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:1"); in Init()
171 flr2_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:2"); in Init()
371 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::__anond2259cec0111::FunctionLibraryRuntimeTest
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 115 pflr_->GetFLR(device_->name()), node_def_, in InitOpWithFunctionLibrary()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 490 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
| D | xla_compiler.cc | 532 pflr_.reset(new ProcessFunctionLibraryRuntime( in XlaCompiler()
537 flib_runtime_ = pflr_->GetFLR(device_->name()); in XlaCompiler()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | multi_device_iterator_ops.cc | 68 pflr_(std::move(pflr)), in MultiDeviceIterator()
380 const std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::data::__anon1c56f74c0111::MultiDeviceIterator
|
| D | dataset_test_base.cc | 395 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in InitFunctionLibraryRuntime()
399 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in InitFunctionLibraryRuntime()
|
| D | dataset_test_base.h | 716 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|