| /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 | 34 class ProcessFunctionLibraryRuntime; variable
72 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| D | process_function_library_runtime.h | 61 class ProcessFunctionLibraryRuntime {
66 ProcessFunctionLibraryRuntime(
75 ~ProcessFunctionLibraryRuntime() { in ~ProcessFunctionLibraryRuntime()
372 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()
269 ProcessFunctionLibraryRuntime* pflr = nullptr) { in Run()
279 ProcessFunctionLibraryRuntime* pflr = nullptr) { in RunWithPackedArgs()
320 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr_;
331 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr( in TEST_F()
332 new ProcessFunctionLibraryRuntime( in TEST_F()
336 proc_flr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST_F()
1198 std::unique_ptr<ProcessFunctionLibraryRuntime> cloned_proc_flr; in TEST_F()
1212 std::unique_ptr<ProcessFunctionLibraryRuntime> cloned_proc_flr; in TEST_F()
|
| D | function.h | 65 ProcessFunctionLibraryRuntime* parent);
|
| D | kernel_benchmark_testlib.cc | 91 pflr_ = std::unique_ptr<ProcessFunctionLibraryRuntime>( in Benchmark()
92 new ProcessFunctionLibraryRuntime( in Benchmark()
|
| D | function.cc | 197 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
316 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr, in Clone()
333 ProcessFunctionLibraryRuntime* parent);
381 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
428 ProcessFunctionLibraryRuntime* parent_ = nullptr; // not owned.
467 ProcessFunctionLibraryRuntime* parent) in FunctionLibraryRuntimeImpl()
478 ? ProcessFunctionLibraryRuntime::kDefaultFLRDevice in FunctionLibraryRuntimeImpl()
1144 ProcessFunctionLibraryRuntime::ReceiveTensorsAsync( in RunRemote()
1176 s = ProcessFunctionLibraryRuntime::SendTensors( in RunRemote()
1423 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr, in Clone()
[all …]
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_compat_request_state.h | 78 const tensorflow::ProcessFunctionLibraryRuntime* pflr);
86 const tensorflow::ProcessFunctionLibraryRuntime* pflr);
98 const tensorflow::ProcessFunctionLibraryRuntime&
156 const tensorflow::ProcessFunctionLibraryRuntime* pflr_;
|
| D | kernel_fallback_compat_request_state.cc | 74 const tensorflow::ProcessFunctionLibraryRuntime* pflr) in KernelFallbackCompatRequestState()
111 const tensorflow::ProcessFunctionLibraryRuntime* pflr) in KernelFallbackCompatRequestState()
|
| D | kernel_fallback_execute_compat.h | 46 const tensorflow::ProcessFunctionLibraryRuntime* pflr,
|
| /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,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | get_compiler_ir.h | 24 class ProcessFunctionLibraryRuntime; variable
42 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()
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | fallback_state.h | 55 const ProcessFunctionLibraryRuntime &process_function_library_runtime() in process_function_library_runtime()
65 ProcessFunctionLibraryRuntime pflr_;
|
| /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 | 339 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
340 std::unique_ptr<ProcessFunctionLibraryRuntime> local_pflr_;
|
| /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
266 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc()
340 ProcessFunctionLibraryRuntime* const pflr_; // non-null
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.h | 116 ProcessFunctionLibraryRuntime* pflr,
123 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| D | standalone.cc | 48 ProcessFunctionLibraryRuntime* pflr, DeviceMgr* device_mgr, in CreateParams()
79 auto pflr = absl::make_unique<ProcessFunctionLibraryRuntime>( in FromGraph()
168 ProcessFunctionLibraryRuntime* pflr, in Dataset()
|
| D | serialization_utils_test.cc | 67 (*result)->pflr_ = absl::make_unique<ProcessFunctionLibraryRuntime>( in Create()
89 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| /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()
|