| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 43 const char ProcessFunctionLibraryRuntime::kDefaultFLRDevice[] = "null";
45 Status ProcessFunctionLibraryRuntime::FunctionData::DistributedInit( in DistributedInit()
58 ProcessFunctionLibraryRuntime::ProcessFunctionLibraryRuntime( in ProcessFunctionLibraryRuntime() function in tensorflow::ProcessFunctionLibraryRuntime
83 ProcessFunctionLibraryRuntime::ProcessFunctionLibraryRuntime( in ProcessFunctionLibraryRuntime() function in tensorflow::ProcessFunctionLibraryRuntime
110 Status ProcessFunctionLibraryRuntime::SendTensors( in SendTensors()
129 void ProcessFunctionLibraryRuntime::ReceiveTensorsAsync( in ReceiveTensorsAsync()
146 Status ProcessFunctionLibraryRuntime::GetDeviceIncarnation( in GetDeviceIncarnation()
156 Status ProcessFunctionLibraryRuntime::GetDeviceContext( in GetDeviceContext()
181 FunctionLibraryRuntime* ProcessFunctionLibraryRuntime::GetFLR( in GetFLR()
198 FunctionLibraryRuntime::Handle ProcessFunctionLibraryRuntime::AddHandle( in AddHandle()
[all …]
|
| D | process_function_library_runtime.h | 29 class ProcessFunctionLibraryRuntime {
34 ProcessFunctionLibraryRuntime(
42 ProcessFunctionLibraryRuntime(const DeviceMgr* device_mgr, Env* env,
220 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr) const;
|
| D | function.h | 61 ProcessFunctionLibraryRuntime* parent);
70 ProcessFunctionLibraryRuntime* parent);
|
| D | process_function_library_runtime_test.cc | 92 proc_flr_.reset(new ProcessFunctionLibraryRuntime( in Init()
216 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr_;
225 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr( in TEST_F()
226 new ProcessFunctionLibraryRuntime( in TEST_F()
230 proc_flr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST_F()
|
| D | function.cc | 204 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
297 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr, in Clone()
313 ProcessFunctionLibraryRuntime* parent);
356 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
398 ProcessFunctionLibraryRuntime* parent_ = nullptr; // not owned.
430 ProcessFunctionLibraryRuntime* parent) in FunctionLibraryRuntimeImpl()
440 ? ProcessFunctionLibraryRuntime::kDefaultFLRDevice in FunctionLibraryRuntimeImpl()
1002 ProcessFunctionLibraryRuntime::ReceiveTensorsAsync( in RunRemote()
1034 s = ProcessFunctionLibraryRuntime::SendTensors( in RunRemote()
1178 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr, in Clone()
[all …]
|
| D | function_threadpool_test.cc | 66 pflr_.reset(new ProcessFunctionLibraryRuntime( in Init()
195 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone.cc | 57 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr = in FromGraph()
58 MakeUnique<ProcessFunctionLibraryRuntime>( in FromGraph()
112 ProcessFunctionLibraryRuntime* pflr, in Dataset()
|
| D | standalone.h | 106 ProcessFunctionLibraryRuntime* pflr,
112 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 50 pflr_ = tensorflow::MakeUnique<ProcessFunctionLibraryRuntime>(
59 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr()
66 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| D | kernel_and_device.h | 41 class ProcessFunctionLibraryRuntime; variable
194 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc()
231 ProcessFunctionLibraryRuntime* const pflr_; // non-null
|
| D | context.h | 96 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr()
279 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| 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/tf2xla/ |
| D | functionalize_control_flow.cc | 238 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in Run()
239 new ProcessFunctionLibraryRuntime( in Run()
243 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
|
| D | xla_compiler.h | 463 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
464 std::unique_ptr<ProcessFunctionLibraryRuntime> local_pflr_;
|
| D | tf2xla_util_test.cc | 292 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in TEST()
293 new ProcessFunctionLibraryRuntime( in TEST()
297 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 61 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in IteratorResource()
149 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in Restore()
227 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in SetIteratorFromDataset()
268 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr, in State()
277 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr, in State()
288 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr;
424 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in Compute()
479 std::unique_ptr<ProcessFunctionLibraryRuntime>* pflr) { in CreatePrivateFLR()
489 *pflr = absl::make_unique<ProcessFunctionLibraryRuntime>( in CreatePrivateFLR()
512 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in Compute()
[all …]
|
| D | graph_rewrite_dataset.h | 84 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_ = nullptr;
|
| D | multi_device_iterator_ops.cc | 50 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in MultiDeviceIterator()
358 const std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
407 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in Compute()
|
| D | iterator_ops.h | 67 std::unique_ptr<ProcessFunctionLibraryRuntime>* pflr);
|
| D | dataset_test_base.h | 187 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op_test.cc | 74 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in Init()
83 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| D | partially_decluster_pass.cc | 286 auto pflr = absl::make_unique<ProcessFunctionLibraryRuntime>( in PartiallyDeclusterGraph()
289 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in PartiallyDeclusterGraph()
|
| D | mark_for_compilation_pass.cc | 453 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in FindCompilationCandidates()
454 new ProcessFunctionLibraryRuntime(nullptr, env, TF_GRAPH_DEF_VERSION, in FindCompilationCandidates()
457 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in FindCompilationCandidates()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.h | 134 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 44 class ProcessFunctionLibraryRuntime; variable
692 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
|