Home
last modified time | relevance | path

Searched refs:instantiate_opts (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime_test.cc100 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in Instantiate() argument
102 return proc_flr_->Instantiate(name, attrs, instantiate_opts, handle); in Instantiate()
156 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in Run() argument
160 proc_flr_->Instantiate(name, attrs, instantiate_opts, &handle); in Run()
273 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
274 instantiate_opts.target = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
278 Run("XTimesTwo", opts, {{"T", DT_FLOAT}}, instantiate_opts, {x}, {&y})); in TEST_F()
289 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
290 instantiate_opts.target = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
292 TF_CHECK_OK(Run("FindDevice", opts, {}, instantiate_opts, {}, {&y})); in TEST_F()
[all …]
Dfunction_test.cc1584 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
1585 instantiate_opts.target = "/device:CPU:1"; in TEST_F()
1587 TF_CHECK_OK(Instantiate(flr0_, "FindDevice", {}, instantiate_opts, &handle)); in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc107 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
108 instantiate_opts.target = "/job:a/replica:0/task:0/device:CPU:0"; in TEST_F()
110 {{"T", DT_FLOAT}}, instantiate_opts, in TEST_F()
312 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
313 instantiate_opts.target = "/job:localhost/replica:0/task:1/cpu:0"; in TEST_F()
318 instantiate_opts, {x}, {&y})); in TEST_F()
327 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
328 instantiate_opts.target = "/job:localhost/replica:0/task:1/cpu:0"; in TEST_F()
333 instantiate_opts, {x1, x2}, {&y1, &y2})); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc319 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in ComputeAsync() local
320 instantiate_opts.target = target_device; in ComputeAsync()
337 instantiate_opts, &handle), in ComputeAsync()