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.cc169 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in Instantiate() argument
171 return proc_flr_->Instantiate(name, attrs, instantiate_opts, handle); in Instantiate()
215 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in RunWithRuntime() argument
219 Status status = pflr->Instantiate(name, attrs, instantiate_opts, &handle); in RunWithRuntime()
272 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in Run() argument
276 name, opts, attrs, instantiate_opts, args, rets, proc_flr_.get()); in Run()
282 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in RunWithPackedArgs() argument
286 name, opts, attrs, instantiate_opts, args, rets, proc_flr_.get()); in RunWithPackedArgs()
384 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
385 instantiate_opts.target = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
[all …]
Dfunction_test.cc495 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
496 instantiate_opts.executor_type = "SINGLE_THREADED_EXECUTOR"; in TEST_F()
500 instantiate_opts, &handle)); in TEST_F()
1976 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
1977 instantiate_opts.target = "/device:CPU:1"; in TEST_F()
1979 TF_CHECK_OK(Instantiate(flr0_, "FindDevice", {}, instantiate_opts, &handle)); in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc125 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
126 instantiate_opts.target = "/job:a/replica:0/task:0/device:CPU:0"; in TEST_F()
128 test::function::Swap().signature(), {{"T", DT_FLOAT}}, instantiate_opts, in TEST_F()
383 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
384 instantiate_opts.target = "/job:localhost/replica:0/task:1/cpu:0"; in TEST_F()
389 instantiate_opts, {x}, {&y})); in TEST_F()
398 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F() local
399 instantiate_opts.target = "/job:localhost/replica:0/task:1/cpu:0"; in TEST_F()
404 instantiate_opts, {x1, x2}, {&y1, &y2})); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc305 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in ComputeAsync() local
310 instantiate_opts.config_proto = *config; in ComputeAsync()
312 instantiate_opts.target = target_device; in ComputeAsync()
315 instantiate_opts, &handle), in ComputeAsync()