| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone.cc | 68 TF_RETURN_IF_ERROR(graph_runner.Run(&graph, pflr->GetFLR("/device:CPU:0"), in FromGraph()
93 op_params.function_library = pflr_->GetFLR("/device:CPU:0"); in MakeIterator()
121 MakeUnique<FunctionHandleCache>(pflr_->GetFLR("/device:CPU:0")); in Dataset()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 148 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceIncarnation()
159 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceContext()
181 FunctionLibraryRuntime* ProcessFunctionLibraryRuntime::GetFLR( in GetFLR() function in tensorflow::ProcessFunctionLibraryRuntime
602 FunctionLibraryRuntime* flr = GetFLR(options.target); in InstantiateMultiDevice()
678 FunctionLibraryRuntime* target_flr = GetFLR(target); in InstantiateMultiDevice()
715 FunctionLibraryRuntime* target_flr = GetFLR(target); in GetOutputDevices()
786 GetFLR(target)->Run( in RunMultiDevice()
814 FunctionLibraryRuntime* flr = GetFLR(options.target); in Instantiate()
874 FunctionLibraryRuntime* flr = GetFLR(device); in ReleaseMultiDeviceHandle()
902 flr = GetFLR(target_device); in ReleaseHandle()
[all …]
|
| D | process_function_library_runtime_test.cc | 230 proc_flr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST_F()
237 proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:0"); in TEST_F()
240 flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0"); in TEST_F()
243 flr = proc_flr_->GetFLR("/device:CPU:0"); in TEST_F()
246 flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:1"); in TEST_F()
249 flr = proc_flr_->GetFLR("abc"); in TEST_F()
|
| D | process_function_library_runtime.h | 80 FunctionLibraryRuntime* GetFLR(const string& device_name) const;
|
| D | function_threadpool_test.cc | 69 flr0_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
|
| D | function_test.cc | 163 flr0_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
164 flr1_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:1"); in Init()
165 flr2_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:2"); in Init()
|
| D | direct_session.cc | 1253 auto lib = func_info->proc_flr->GetFLR(partition_name); in CreateExecutors()
|
| D | function.cc | 1183 *out_flr = (*out_pflr)->GetFLR(device_->name()); in Clone()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op_test.cc | 77 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
|
| D | partially_decluster_pass.cc | 289 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in PartiallyDeclusterGraph()
|
| D | mark_for_compilation_pass.cc | 457 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in FindCompilationCandidates()
|
| D | extract_outside_compilation_pass_test.cc | 251 auto flr = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in ExtractOutsideCompilationTest()
|
| D | encapsulate_subgraphs_pass.cc | 2568 pflr->GetFLR("/job:localhost/replica:0/task:0/device:CPU:0"); in Run()
|
| D | encapsulate_subgraphs_pass_test.cc | 514 auto flr = pflr->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Encapsulate()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 54 flr_ = pflr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0");
|
| D | context.h | 93 return pflr_->GetFLR(d->name()); in func_lib()
|
| D | kernel_and_device.cc | 88 function_def = pflr_->GetFLR(host_cpu_device_->name()) in Init()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 243 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
|
| D | tf2xla_util_test.cc | 297 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST()
|
| D | xla_compiler.cc | 445 local_flib_runtime_ = local_pflr_->GetFLR(device_->name()); in XlaCompiler()
446 flib_runtime_ = pflr_->GetFLR(device_->name()); in XlaCompiler()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 161 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in InitFunctionLibraryRuntime()
|
| D | iterator_ops.cc | 494 return (*pflr)->GetFLR(ctx->device()->name()); in CreatePrivateFLR()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 227 FunctionLibraryRuntime* lib = item->proc_flr->GetFLR(unit->device->name()); in InitItem()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 133 FunctionLibraryRuntime* flr = pflr->GetFLR(cpu_device->name()); in OptimizeGraph()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 420 flr_ = process_flr_->GetFLR(device_mgr_->ListDevices()[0]->name()); in InitializeFunctionLibraryRuntime()
|