Home
last modified time | relevance | path

Searched refs:InstantiateOptions (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime_test.cc48 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
100 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in Instantiate()
156 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in Run()
273 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
289 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
306 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
324 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
345 FunctionLibraryRuntime::InstantiateOptions instantiate_opts_0; in TEST_F()
351 FunctionLibraryRuntime::InstantiateOptions instantiate_opts_1; in TEST_F()
366 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
[all …]
Dprocess_function_library_runtime.h121 const FunctionLibraryRuntime::InstantiateOptions& options,
231 const FunctionLibraryRuntime::InstantiateOptions& options,
275 const FunctionLibraryRuntime::InstantiateOptions& options);
Dfunction_threadpool_test.cc118 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
128 FunctionLibraryRuntime::InstantiateOptions(), args, in InstantiateAndRun()
135 const FunctionLibraryRuntime::InstantiateOptions& options, in InstantiateAndRun()
Dfunction_test.cc215 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
225 FunctionLibraryRuntime::InstantiateOptions(), args, in InstantiateAndRun()
232 const FunctionLibraryRuntime::InstantiateOptions& options, in InstantiateAndRun()
414 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
469 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
545 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
560 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
575 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
619 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
628 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
[all …]
Dfunction.cc175 const InstantiateOptions& options,
216 const InstantiateOptions& options, Handle* handle) { in Instantiate()
220 InstantiateOptions options_copy = options; in Instantiate()
318 const InstantiateOptions& options,
411 bool IsLocalTarget(const InstantiateOptions& options);
563 InstantiateOptions options; in CreateKernel()
632 InstantiateOptions options; in InstantiateSymbolicGradient()
647 const InstantiateOptions& options) { in IsLocalTarget()
668 const InstantiateOptions& options, Handle* handle) { in Instantiate()
675 InstantiateOptions options_copy(options); in Instantiate()
Dprocess_function_library_runtime.cc48 const FunctionLibraryRuntime::InstantiateOptions& options) { in DistributedInit()
428 const FunctionLibraryRuntime::InstantiateOptions& options) { in ValidateMultiDeviceOptions()
514 const FunctionLibraryRuntime::InstantiateOptions& options, in InstantiateMultiDevice()
680 FunctionLibraryRuntime::InstantiateOptions opts; in InstantiateMultiDevice()
807 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc53 const FunctionLibraryRuntime::InstantiateOptions& options, GraphDef* g, in ConstructFunctionGraphHelper()
62 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
71 const FunctionLibraryRuntime::InstantiateOptions& options, in InstantiateAndRun()
107 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
312 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
327 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
Dcluster_function_library_runtime.h39 const FunctionLibraryRuntime::InstantiateOptions& options,
50 const FunctionLibraryRuntime::InstantiateOptions& options, GraphDef* g,
Dcluster_function_library_runtime.cc30 const FunctionLibraryRuntime::InstantiateOptions& options, GraphDef* g, in ConstructFunctionGraph()
122 AttrSlice attrs, const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
/external/tensorflow/tensorflow/core/framework/
Dfunction.h506 struct InstantiateOptions { struct
587 const InstantiateOptions& options, argument
699 static string ExecutorType(const InstantiateOptions& options,
711 const FunctionLibraryRuntime::InstantiateOptions& options);
731 const FunctionLibraryRuntime::InstantiateOptions& options,
Dfunction_handle_cache.h35 FunctionLibraryRuntime::InstantiateOptions options,
Dfunction_handle_cache.cc36 FunctionLibraryRuntime::InstantiateOptions options, in Instantiate()
Dfunction.cc877 string FunctionLibraryRuntime::ExecutorType(const InstantiateOptions& options, in ExecutorType()
889 const FunctionLibraryRuntime::InstantiateOptions& options) { in Canonicalize()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.h46 FunctionLibraryRuntime::InstantiateOptions* opts);
Dpartitioned_function_ops.cc130 AttrSlice attrs, FunctionLibraryRuntime::InstantiateOptions* opts) { in FillOutputDevices()
170 FunctionLibraryRuntime::InstantiateOptions opts; in Instantiate()
Dfunction_ops.cc319 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in ComputeAsync()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc103 FunctionLibraryRuntime::InstantiateOptions options; in Init()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc136 FunctionLibraryRuntime::InstantiateOptions inst_opts; in Instantiate()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc172 FunctionLibraryRuntime::InstantiateOptions inst_ops; in ConstructFunctionHandle()