| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 54 const char ProcessFunctionLibraryRuntime::kDefaultFLRDevice[] = "null";
56 void ProcessFunctionLibraryRuntime::FunctionData::DistributedInit( in DistributedInit()
78 ProcessFunctionLibraryRuntime::ProcessFunctionLibraryRuntime( in ProcessFunctionLibraryRuntime() function in tensorflow::ProcessFunctionLibraryRuntime
121 Status ProcessFunctionLibraryRuntime::SendTensors( in SendTensors()
140 void ProcessFunctionLibraryRuntime::ReceiveTensorsAsync( in ReceiveTensorsAsync()
158 Status ProcessFunctionLibraryRuntime::GetRetTypes( in GetRetTypes()
179 Status ProcessFunctionLibraryRuntime::GetDeviceIncarnation( in GetDeviceIncarnation()
189 Status ProcessFunctionLibraryRuntime::GetDeviceContext( in GetDeviceContext()
214 FunctionLibraryRuntime* ProcessFunctionLibraryRuntime::GetFLR( in GetFLR()
231 FunctionLibraryRuntime::Handle ProcessFunctionLibraryRuntime::AddHandle( in AddHandle()
[all …]
|
| D | process_function_library_runtime.h | 60 class ProcessFunctionLibraryRuntime {
65 ProcessFunctionLibraryRuntime(
74 virtual ~ProcessFunctionLibraryRuntime() { in ~ProcessFunctionLibraryRuntime()
350 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
|
| D | process_function_library_runtime_test.cc | 120 proc_flr_.reset(new ProcessFunctionLibraryRuntime( in Init()
189 ProcessFunctionLibraryRuntime* pflr) { in RunWithRuntime()
289 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr_;
298 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr( in TEST_F()
299 new ProcessFunctionLibraryRuntime( in TEST_F()
303 proc_flr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST_F()
1020 std::unique_ptr<ProcessFunctionLibraryRuntime> cloned_proc_flr; in TEST_F()
|
| D | function_threadpool_test.cc | 66 pflr_.reset(new ProcessFunctionLibraryRuntime( in Init()
194 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| D | function.cc | 208 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
315 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr, in Clone()
333 ProcessFunctionLibraryRuntime* parent);
376 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
423 ProcessFunctionLibraryRuntime* parent_ = nullptr; // not owned.
459 ProcessFunctionLibraryRuntime* parent) in FunctionLibraryRuntimeImpl()
471 ? ProcessFunctionLibraryRuntime::kDefaultFLRDevice in FunctionLibraryRuntimeImpl()
1068 ProcessFunctionLibraryRuntime::ReceiveTensorsAsync( in RunRemote()
1100 s = ProcessFunctionLibraryRuntime::SendTensors( in RunRemote()
1254 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr, in Clone()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 39 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in IteratorResource()
74 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr, in State()
93 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr;
130 std::unique_ptr<ProcessFunctionLibraryRuntime>* pflr);
154 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr,
|
| D | cache_ops.h | 68 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr,
|
| D | random_seed_ops.h | 66 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr,
|
| D | cache_ops.cc | 77 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in CreateResource()
|
| D | dataset_utils.h | 85 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in Compute()
111 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr,
|
| D | multi_device_iterator_ops.cc | 59 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in MultiDeviceIterator()
380 const std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
428 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in Compute()
524 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in CreateResource()
|
| D | random_seed_ops.cc | 103 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in CreateResource()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | process_function_library_runtime.h | 40 : public ProcessFunctionLibraryRuntime {
49 : ProcessFunctionLibraryRuntime( in ProcessFunctionLibraryRuntime() function
|
| D | kernel_and_device_test.cc | 51 pflr_ = tensorflow::MakeUnique<ProcessFunctionLibraryRuntime>(
61 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr()
68 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| D | kernel_and_device.h | 49 class ProcessFunctionLibraryRuntime; variable
232 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc()
293 ProcessFunctionLibraryRuntime* const pflr_; // non-null
|
| D | process_function_library_runtime.cc | 61 return ProcessFunctionLibraryRuntime::Run(opts, handle, args, rets, in Run()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 226 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in Run()
227 new ProcessFunctionLibraryRuntime( in Run()
232 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
|
| D | xla_compiler.h | 490 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
491 std::unique_ptr<ProcessFunctionLibraryRuntime> local_pflr_;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator_test.cc | 75 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in Init()
85 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| D | xla_cluster_util_test.cc | 150 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in GetNodesRelatedToRefVarsSorted()
151 new ProcessFunctionLibraryRuntime( in GetNodesRelatedToRefVarsSorted()
155 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in GetNodesRelatedToRefVarsSorted()
|
| 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_;
|
| D | partially_decluster_pass.cc | 298 auto pflr = absl::make_unique<ProcessFunctionLibraryRuntime>( in PartiallyDeclusterGraph()
301 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in PartiallyDeclusterGraph()
|
| /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()
|
| D | ops_testutil.h | 185 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.h | 121 friend class ProcessFunctionLibraryRuntime; variable
|