| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 85 TF_RETURN_IF_ERROR(graph_runner.Run(&graph, pflr->GetFLR("/device:CPU:0"), in FromGraph()
111 op_params.function_library = pflr_->GetFLR("/device:CPU:0"); in MakeIterator()
154 absl::make_unique<FunctionHandleCache>(pflr_->GetFLR("/device:CPU:0")); in Dataset()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 166 flr = GetFLR(fiter->second->target_device()); in GetRetTypes()
177 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceIncarnation()
188 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceContext()
233 FunctionLibraryRuntime* ProcessFunctionLibraryRuntime::GetFLR( in GetFLR() function in tensorflow::ProcessFunctionLibraryRuntime
733 FunctionLibraryRuntime* flr = GetFLR(options.target); in InstantiateMultiDevice()
990 FunctionLibraryRuntime* flr = GetFLR(opts.target); in InstantiateMultiDevice()
1033 FunctionLibraryRuntime* target_flr = GetFLR(target); in GetOutputDevices()
1170 FunctionLibraryRuntime* flr = GetFLR(target); in RunMultiDevice()
1210 FunctionLibraryRuntime* flr = GetFLR(options.target); in Instantiate()
1307 FunctionLibraryRuntime* flr = GetFLR(device); in ReleaseMultiDeviceHandle()
[all …]
|
| D | process_function_library_runtime_test.cc | 344 proc_flr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST_F()
351 proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:0"); in TEST_F()
354 flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0"); in TEST_F()
357 flr = proc_flr_->GetFLR("/device:CPU:0"); in TEST_F()
360 flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:1"); in TEST_F()
363 flr = proc_flr_->GetFLR("abc"); in TEST_F()
1203 auto* flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:0"); in TEST_F()
1217 auto* flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/cpu:0"); in TEST_F()
|
| D | function_threadpool_test.cc | 74 flr0_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
|
| D | kernel_benchmark_testlib.cc | 98 flr_ = pflr_->GetFLR(device_->name()); in Benchmark()
|
| D | process_function_library_runtime.h | 114 FunctionLibraryRuntime* GetFLR(const string& device_name) const;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | force_xla_constants_on_host_pass.cc | 33 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
|
| D | xla_kernel_creator_test.cc | 81 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
|
| D | get_compiler_ir.cc | 56 FunctionLibraryRuntime* flr = pflr->GetFLR(dev->name()); in GetCompilerIr()
|
| D | partially_decluster_pass.cc | 301 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in PartiallyDeclusterGraph()
|
| D | xla_cluster_util_test.cc | 155 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in GetNodesRelatedToRefVarsSorted()
|
| D | compilability_check_util_test.cc | 102 return pflr_->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in GetFunctionLibraryRuntime()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 290 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in FunctionalizeControlFlow()
340 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
|
| D | const_analysis_test.cc | 131 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TestFunctionCall()
|
| D | resource_util_test.cc | 61 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in AnalyzeAndVerify()
|
| D | tf2xla_util_test.cc | 302 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 56 flr_ = pflr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0");
|
| D | kernel_and_device.cc | 142 function_def = pflr_->GetFLR(host_cpu_device_->name()) in InstantiateFunc()
|
| D | context.h | 150 return pflr_->GetFLR(d->name()); in func_lib()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 139 auto flr = pflr_->GetFLR(device_->name()); in InitOpWithFunctionLibrary()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 178 params_->function_library = pflr_->GetFLR(device_->name()); in RunOpKernel()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 947 FunctionLibraryRuntime* flr = eager_pflr_->GetFLR(remote_device_); in TEST_F()
995 FunctionLibraryRuntime* flr = eager_pflr_->GetFLR(remote_device_); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 268 FunctionLibraryRuntime* flr = pflr->GetFLR(cpu_device->name()); in RuntimeGraphOptimizer()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 239 FunctionLibraryRuntime* lib = item->proc_flr->GetFLR(unit->device->name()); in InitItem()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 399 params_.function_library = pflr_->GetFLR(device_->name()); in PrepareParams()
|