/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()
|