| /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::__anon9d2e75500111::TestEnv
|
| D | kernel_and_device.cc | 91 Status status = pflr_->ReleaseHandle(handle_); in ~KernelAndDeviceFunc()
142 function_def = pflr_->GetFLR(host_cpu_device_->name()) in InstantiateFunc()
167 pflr_->GetFunctionLibraryDefinition(); in InstantiateFunc()
219 pflr_->Instantiate(ndef.op(), AttrSlice(ndef), options, &handle_)); in InstantiateFunc()
220 return pflr_->IsCrossProcess(handle_, &is_cross_process_); in InstantiateFunc()
228 return pflr_->GetOutputDevices(handle_, &output_devices_); in Init()
393 pflr_->Run(*opts, handle_, inputs, outputs, in RunAsync()
|
| D | context.cc | 196 pflr_.reset(new ProcessFunctionLibraryRuntime( in ResetPFLR()
948 pflr_->AddCompositeDevice(*composite_device); in FindOrCreateCompositeDevice()
1098 const ConfigProto* config = pflr_ ? pflr_->config() : nullptr; in StoreCollectiveOpsServer()
1272 pflr_->InitializeDeviceAndFlr(); in UpdateRemoteMaster()
1353 const auto* config = pflr_->config(); in SetMasterContextState()
1455 const auto* config = pflr_->config(); in InitializeRemoteWorker()
1491 pflr_->InitializeDeviceAndFlr(); in UpdateRemoteWorker()
|
| D | kernel_and_device.h | 274 pflr_(pflr), in KernelAndDeviceFunc()
326 ProcessFunctionLibraryRuntime* const pflr_; // non-null
|
| D | context.h | 150 return pflr_->GetFLR(d->name()); in func_lib()
153 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr()
614 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator_test.cc | 77 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in Init()
81 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
87 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::XlaKernelCreatorTest
|
| D | compilability_check_util_test.cc | 98 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in GetFunctionLibraryRuntime()
102 return pflr_->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in GetFunctionLibraryRuntime()
113 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::__anon66c333b40111::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/data/ |
| D | standalone.cc | 111 op_params.function_library = pflr_->GetFLR("/device:CPU:0"); in MakeIterator()
150 pflr_(pflr), in Dataset()
154 absl::make_unique<FunctionHandleCache>(pflr_->GetFLR("/device:CPU:0")); in Dataset()
|
| D | standalone.h | 118 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 85 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in OpsTestBase()
109 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in SetDevice()
178 params_->function_library = pflr_->GetFLR(device_->name()); in RunOpKernel()
|
| D | ops_testutil.h | 186 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 84 pflr_(std::move(pflr)), in State()
100 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr() { return pflr_; } in pflr()
117 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
| D | multi_device_iterator_ops.cc | 68 pflr_(std::move(pflr)), in MultiDeviceIterator()
382 const std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::data::__anond56f2ce10111::MultiDeviceIterator
|
| D | dataset_test_base.cc | 402 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in InitFunctionLibraryRuntime()
412 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in InitFunctionLibraryRuntime()
|
| D | dataset_test_base.h | 763 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_threadpool_test.cc | 65 pflr_.reset(new ProcessFunctionLibraryRuntime( in Init()
74 flr0_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
199 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::__anon49279c120111::FunctionLibraryRuntimeTest
|
| D | kernel_benchmark_testlib.cc | 92 pflr_ = std::unique_ptr<ProcessFunctionLibraryRuntime>( in Benchmark()
98 flr_ = pflr_->GetFLR(device_->name()); in Benchmark()
135 pflr_.reset(); in ~Benchmark()
|
| D | kernel_benchmark_testlib.h | 95 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
| D | function_test.cc | 162 pflr_.reset(new ProcessFunctionLibraryRuntime( in Init()
171 flr0_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
172 flr1_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:1"); in Init()
173 flr2_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:2"); in Init()
373 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; member in tensorflow::__anon3370e4410111::FunctionLibraryRuntimeTest
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 350 std::unique_ptr<tensorflow::ProcessFunctionLibraryRuntime> pflr_; member in mlir::mhlo::__anon6f7e6d590111::Tf2XlaRewriter
396 pflr_ = absl::make_unique<tensorflow::ProcessFunctionLibraryRuntime>( in PrepareParams()
399 params_.function_library = pflr_->GetFLR(device_->name()); in PrepareParams()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 336 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; variable
|
| D | xla_compiler.cc | 518 pflr_.reset(new ProcessFunctionLibraryRuntime( in XlaCompiler()
523 flib_runtime_ = pflr_->GetFLR(device_->name()); in XlaCompiler()
1295 pflr_->GetFunctionLibraryDefinition())); in CompileGraph()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 139 auto flr = pflr_->GetFLR(device_->name()); in InitOpWithFunctionLibrary()
|