Home
last modified time | relevance | path

Searched refs:ProcessFunctionLibraryRuntime (Results 1 – 25 of 75) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dprocess_function_library_runtime.cc64 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 Dprocess_function_library_runtime_test.cc141 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 Dfunction.cc198 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 Dprocess_function_library_runtime.h62 class ProcessFunctionLibraryRuntime {
67 ProcessFunctionLibraryRuntime(
76 ~ProcessFunctionLibraryRuntime() { in ~ProcessFunctionLibraryRuntime()
381 std::unique_ptr<ProcessFunctionLibraryRuntime>* out_pflr,
H A Dkernel_benchmark_testlib.h34 class ProcessFunctionLibraryRuntime; variable
72 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
H A Dkernel_benchmark_testlib.cc91 pflr_ = std::unique_ptr<ProcessFunctionLibraryRuntime>( in Benchmark()
92 new ProcessFunctionLibraryRuntime( in Benchmark()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Diterator_ops.h43 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 Diterator_ops.cc94 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 Dmulti_device_iterator_ops.cc82 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 Dkernel_fallback_compat_request_state.h84 const tensorflow::ProcessFunctionLibraryRuntime* pflr);
94 const tensorflow::ProcessFunctionLibraryRuntime* pflr);
110 const tensorflow::ProcessFunctionLibraryRuntime&
168 const tensorflow::ProcessFunctionLibraryRuntime* pflr_ = nullptr;
H A Dkernel_fallback_compat_request_state.cc70 const tensorflow::ProcessFunctionLibraryRuntime* pflr) in KernelFallbackCompatRequestState()
108 const tensorflow::ProcessFunctionLibraryRuntime* pflr) in KernelFallbackCompatRequestState()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfunctionalize_control_flow.cc282 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 Dconst_analysis_test.cc126 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in TestFunctionCall()
127 new ProcessFunctionLibraryRuntime(nullptr, Env::Default(), in TestFunctionCall()
131 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TestFunctionCall()
H A Dxla_compiler.h348 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
349 std::unique_ptr<ProcessFunctionLibraryRuntime> local_pflr_;
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device_test.cc51 pflr_ = tensorflow::MakeUnique<ProcessFunctionLibraryRuntime>(
61 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr()
68 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
H A Dkernel_and_device.h53 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 Dstandalone.cc48 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 Dstandalone.h116 DeviceMgr* device_mgr, ProcessFunctionLibraryRuntime* pflr,
124 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dget_compiler_ir.h24 class ProcessFunctionLibraryRuntime; variable
43 IrExportStage stage, ProcessFunctionLibraryRuntime* pflr,
H A Dforce_xla_constants_on_host_pass.cc29 auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>( in Run()
33 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
H A Dxla_cluster_util_test.cc150 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in GetNodesRelatedToRefVarsSorted()
151 new ProcessFunctionLibraryRuntime( in GetNodesRelatedToRefVarsSorted()
155 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in GetNodesRelatedToRefVarsSorted()
H A Dxla_kernel_creator_test.cc77 pflr_ = std::make_unique<ProcessFunctionLibraryRuntime>( in Init()
87 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/
H A Dfallback_state.h56 const ProcessFunctionLibraryRuntime &process_function_library_runtime() in process_function_library_runtime()
66 ProcessFunctionLibraryRuntime pflr_;
H A Dop_kernel_runner.h46 const tensorflow::ProcessFunctionLibraryRuntime&
52 const tensorflow::ProcessFunctionLibraryRuntime&
H A Dop_kernel_runner.cc85 const tensorflow::ProcessFunctionLibraryRuntime& in Create()
105 const tensorflow::ProcessFunctionLibraryRuntime& in Create()

123