| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 61 const char ProcessFunctionLibraryRuntime::kDefaultFLRDevice[] = "null";
63 void ProcessFunctionLibraryRuntime::FunctionData::DistributedInit( in DistributedInit()
85 ProcessFunctionLibraryRuntime::ProcessFunctionLibraryRuntime( in ProcessFunctionLibraryRuntime() function in tensorflow::ProcessFunctionLibraryRuntime
117 Status ProcessFunctionLibraryRuntime::SendTensors( in SendTensors()
136 void ProcessFunctionLibraryRuntime::ReceiveTensorsAsync( in ReceiveTensorsAsync()
154 Status ProcessFunctionLibraryRuntime::GetRetTypes( in GetRetTypes()
175 Status ProcessFunctionLibraryRuntime::GetDeviceIncarnation( in GetDeviceIncarnation()
185 Status ProcessFunctionLibraryRuntime::GetDeviceContext( in GetDeviceContext()
212 void ProcessFunctionLibraryRuntime::InitializeDeviceAndFlr() { in InitializeDeviceAndFlr()
233 FunctionLibraryRuntime* ProcessFunctionLibraryRuntime::GetFLR( in GetFLR()
[all …]
|
| D | kernel_benchmark_testlib.h | 39 class ProcessFunctionLibraryRuntime; variable
95 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| D | process_function_library_runtime.h | 61 class ProcessFunctionLibraryRuntime {
66 ProcessFunctionLibraryRuntime(
75 ~ProcessFunctionLibraryRuntime() { in ~ProcessFunctionLibraryRuntime()
360 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
|
| D | process_function_library_runtime_test.cc | 139 proc_flr_.reset(new ProcessFunctionLibraryRuntime( in Init()
217 ProcessFunctionLibraryRuntime* pflr) { in RunWithRuntime()
274 ProcessFunctionLibraryRuntime* pflr = nullptr) { in Run()
284 ProcessFunctionLibraryRuntime* pflr = nullptr) { in RunWithPackedArgs()
328 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr_;
339 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr( in TEST_F()
340 new ProcessFunctionLibraryRuntime( in TEST_F()
344 proc_flr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST_F()
1206 std::unique_ptr<ProcessFunctionLibraryRuntime> cloned_proc_flr; in TEST_F()
1220 std::unique_ptr<ProcessFunctionLibraryRuntime> cloned_proc_flr; in TEST_F()
|
| D | function.h | 65 ProcessFunctionLibraryRuntime* parent);
|
| D | function.cc | 194 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
313 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr, in Clone()
330 ProcessFunctionLibraryRuntime* parent);
378 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
425 ProcessFunctionLibraryRuntime* parent_ = nullptr; // not owned.
464 ProcessFunctionLibraryRuntime* parent) in FunctionLibraryRuntimeImpl()
475 ? ProcessFunctionLibraryRuntime::kDefaultFLRDevice in FunctionLibraryRuntimeImpl()
1050 ProcessFunctionLibraryRuntime::ReceiveTensorsAsync( in RunRemote()
1082 s = ProcessFunctionLibraryRuntime::SendTensors( in RunRemote()
1327 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr, in Clone()
[all …]
|
| D | function_threadpool_test.cc | 65 pflr_.reset(new ProcessFunctionLibraryRuntime( in Init()
199 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| D | kernel_benchmark_testlib.cc | 92 pflr_ = std::unique_ptr<ProcessFunctionLibraryRuntime>( in Benchmark()
93 new ProcessFunctionLibraryRuntime( in Benchmark()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 39 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr,
79 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr, in State()
100 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr() { return pflr_; } in pflr()
117 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr_;
162 std::unique_ptr<ProcessFunctionLibraryRuntime>* pflr);
186 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr,
|
| D | cache_ops.h | 83 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr,
|
| D | cache_ops.cc | 79 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in CreateResource()
|
| D | random_seed_ops.cc | 85 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in CreateResource()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | get_compiler_ir.h | 24 class ProcessFunctionLibraryRuntime; variable
41 IrExportStage stage, ProcessFunctionLibraryRuntime* pflr,
|
| D | force_xla_constants_on_host_pass.cc | 29 auto pflr = absl::make_unique<ProcessFunctionLibraryRuntime>( in Run()
33 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
|
| D | xla_kernel_creator_test.cc | 77 pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in Init()
87 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 | partially_decluster_pass.cc | 298 auto pflr = absl::make_unique<ProcessFunctionLibraryRuntime>( in PartiallyDeclusterGraph()
301 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in PartiallyDeclusterGraph()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| 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 | 53 class ProcessFunctionLibraryRuntime; variable
261 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc()
326 ProcessFunctionLibraryRuntime* const pflr_; // non-null
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 283 auto pflr = absl::make_unique<ProcessFunctionLibraryRuntime>( in FunctionalizeControlFlow()
290 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in FunctionalizeControlFlow()
334 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in Run()
335 new ProcessFunctionLibraryRuntime( in Run()
340 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
|
| D | const_analysis_test.cc | 126 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in TestFunctionCall()
127 new ProcessFunctionLibraryRuntime(nullptr, Env::Default(), in TestFunctionCall()
131 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TestFunctionCall()
|
| D | xla_compiler.h | 336 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
337 std::unique_ptr<ProcessFunctionLibraryRuntime> local_pflr_;
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.h | 112 ProcessFunctionLibraryRuntime* pflr,
118 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| D | standalone.cc | 58 auto pflr = absl::make_unique<ProcessFunctionLibraryRuntime>( in FromGraph()
145 ProcessFunctionLibraryRuntime* pflr, in Dataset()
|
| /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()
|