| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | process_function_library_runtime.cc | 64 const char ProcessFunctionLibraryRuntime::kDefaultFLRDevice[] = "null";
66 void ProcessFunctionLibraryRuntime::FunctionData::DistributedInit( in DistributedInit()
88 ProcessFunctionLibraryRuntime::ProcessFunctionLibraryRuntime( in ProcessFunctionLibraryRuntime() function in tensorflow::ProcessFunctionLibraryRuntime
120 Status ProcessFunctionLibraryRuntime::SendTensors( in SendTensors()
139 void ProcessFunctionLibraryRuntime::ReceiveTensorsAsync( in ReceiveTensorsAsync()
157 Status ProcessFunctionLibraryRuntime::GetRetTypes( in GetRetTypes()
178 Status ProcessFunctionLibraryRuntime::GetDeviceIncarnation( in GetDeviceIncarnation()
188 Status ProcessFunctionLibraryRuntime::GetDeviceContext( in GetDeviceContext()
215 void ProcessFunctionLibraryRuntime::InitializeDeviceAndFlr() { in InitializeDeviceAndFlr()
252 FunctionLibraryRuntime* ProcessFunctionLibraryRuntime::GetFLR( in GetFLR()
[all …]
|
| H A D | process_function_library_runtime_test.cc | 141 proc_flr_.reset(new ProcessFunctionLibraryRuntime( in Init()
219 ProcessFunctionLibraryRuntime* pflr) { in RunWithRuntime()
271 ProcessFunctionLibraryRuntime* pflr = nullptr) { in Run()
281 ProcessFunctionLibraryRuntime* pflr = nullptr) { in RunWithPackedArgs()
322 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr_;
333 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr( in TEST_F()
334 new ProcessFunctionLibraryRuntime( in TEST_F()
338 proc_flr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST_F()
352 std::unique_ptr<ProcessFunctionLibraryRuntime> proc_flr( in TEST_F()
353 new ProcessFunctionLibraryRuntime( in TEST_F()
[all …]
|
| H A D | function.cc | 198 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
317 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr, in Clone()
334 ProcessFunctionLibraryRuntime* parent);
382 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
431 ProcessFunctionLibraryRuntime* parent_ = nullptr; // not owned.
472 ProcessFunctionLibraryRuntime* parent) in FunctionLibraryRuntimeImpl()
483 ? ProcessFunctionLibraryRuntime::kDefaultFLRDevice in FunctionLibraryRuntimeImpl()
1085 ProcessFunctionLibraryRuntime::ReceiveTensorsAsync( in RunRemote()
1117 s = ProcessFunctionLibraryRuntime::SendTensors( in RunRemote()
1364 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr, in Clone()
[all …]
|
| H A D | process_function_library_runtime.h | 62 class ProcessFunctionLibraryRuntime {
67 ProcessFunctionLibraryRuntime(
76 ~ProcessFunctionLibraryRuntime() { in ~ProcessFunctionLibraryRuntime()
381 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
|
| H A D | kernel_benchmark_testlib.h | 34 class ProcessFunctionLibraryRuntime; variable
72 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| H A D | kernel_benchmark_testlib.cc | 91 pflr_ = std::unique_ptr<ProcessFunctionLibraryRuntime>( in Benchmark()
92 new ProcessFunctionLibraryRuntime( in Benchmark()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | iterator_ops.h | 43 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr,
84 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr, in State()
110 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr() { return pflr_; } in pflr()
129 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr_;
169 std::unique_ptr<ProcessFunctionLibraryRuntime>* pflr);
193 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr,
|
| H A D | iterator_ops.cc | 94 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in IteratorResource()
469 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in Compute()
524 std::unique_ptr<ProcessFunctionLibraryRuntime>* pflr) { in CreatePrivateFLR()
537 *pflr = std::make_unique<ProcessFunctionLibraryRuntime>( in CreatePrivateFLR()
572 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in CreateResource()
789 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in TryInit()
|
| H A D | multi_device_iterator_ops.cc | 82 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in MultiDeviceIterator()
466 const std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
515 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in Compute()
615 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in CreateResource()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | kernel_fallback_compat_request_state.h | 84 const tensorflow::ProcessFunctionLibraryRuntime* pflr);
94 const tensorflow::ProcessFunctionLibraryRuntime* pflr);
110 const tensorflow::ProcessFunctionLibraryRuntime&
168 const tensorflow::ProcessFunctionLibraryRuntime* pflr_ = nullptr;
|
| H A D | kernel_fallback_compat_request_state.cc | 70 const tensorflow::ProcessFunctionLibraryRuntime* pflr) in KernelFallbackCompatRequestState()
108 const tensorflow::ProcessFunctionLibraryRuntime* pflr) in KernelFallbackCompatRequestState()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_control_flow.cc | 282 auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>( in FunctionalizeControlFlow()
289 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in FunctionalizeControlFlow()
333 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in Run()
334 new ProcessFunctionLibraryRuntime( in Run()
339 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
|
| H A 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()
|
| H A D | xla_compiler.h | 348 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
349 std::unique_ptr<ProcessFunctionLibraryRuntime> local_pflr_;
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A 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_;
|
| H A D | kernel_and_device.h | 53 class ProcessFunctionLibraryRuntime; variable
271 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc()
355 ProcessFunctionLibraryRuntime* const pflr_; // non-null
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | standalone.cc | 48 ProcessFunctionLibraryRuntime* pflr, DeviceMgr* device_mgr, in CreateParams()
79 auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>( in FromGraph()
168 DeviceMgr* device_mgr, ProcessFunctionLibraryRuntime* pflr, in Dataset()
|
| H A D | standalone.h | 116 DeviceMgr* device_mgr, ProcessFunctionLibraryRuntime* pflr,
124 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | get_compiler_ir.h | 24 class ProcessFunctionLibraryRuntime; variable
43 IrExportStage stage, ProcessFunctionLibraryRuntime* pflr,
|
| H A D | force_xla_constants_on_host_pass.cc | 29 auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>( in Run()
33 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
|
| H A 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()
|
| H A D | xla_kernel_creator_test.cc | 77 pflr_ = std::make_unique<ProcessFunctionLibraryRuntime>( in Init()
87 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/ |
| H A D | fallback_state.h | 56 const ProcessFunctionLibraryRuntime &process_function_library_runtime() in process_function_library_runtime()
66 ProcessFunctionLibraryRuntime pflr_;
|
| H A D | op_kernel_runner.h | 46 const tensorflow::ProcessFunctionLibraryRuntime&
52 const tensorflow::ProcessFunctionLibraryRuntime&
|
| H A D | op_kernel_runner.cc | 85 const tensorflow::ProcessFunctionLibraryRuntime& in Create()
105 const tensorflow::ProcessFunctionLibraryRuntime& in Create()
|