Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime_test.cc54 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
130 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in Instantiate()
187 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in RunWithRuntime()
244 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in Run()
344 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
359 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
375 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
392 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
412 FunctionLibraryRuntime::InstantiateOptions instantiate_opts_0; in TEST_F()
418 FunctionLibraryRuntime::InstantiateOptions instantiate_opts_1; in TEST_F()
[all …]
Dprocess_function_library_runtime.h158 const FunctionLibraryRuntime::InstantiateOptions& options,
357 const FunctionLibraryRuntime::InstantiateOptions& options,
362 const FunctionLibraryRuntime::InstantiateOptions& options,
414 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.cc221 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
231 FunctionLibraryRuntime::InstantiateOptions(), args, in InstantiateAndRun()
238 const FunctionLibraryRuntime::InstantiateOptions& options, in InstantiateAndRun()
419 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
471 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
549 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
564 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
579 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
622 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
631 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
[all …]
Dfunction.cc175 const InstantiateOptions& options,
221 const InstantiateOptions& options, Handle* handle) { in Instantiate()
225 InstantiateOptions options_copy = options; in Instantiate()
338 const InstantiateOptions& options,
439 bool IsLocalTarget(const InstantiateOptions& options) const;
624 InstantiateOptions options; in CreateKernel()
695 InstantiateOptions options; in InstantiateSymbolicGradient()
710 const InstantiateOptions& options) const { in IsLocalTarget()
731 const InstantiateOptions& options, Handle* handle) { in Instantiate()
738 InstantiateOptions options_copy(options); in Instantiate()
Dprocess_function_library_runtime.cc59 const FunctionLibraryRuntime::InstantiateOptions& options, in DistributedInit()
519 const FunctionLibraryRuntime::InstantiateOptions& options) { in ValidateMultiDeviceOptions()
592 const FunctionLibraryRuntime::InstantiateOptions& options, in InstantiateMultiDevice()
838 FunctionLibraryRuntime::InstantiateOptions opts; in InstantiateMultiDevice()
1080 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
1121 const FunctionLibraryRuntime::InstantiateOptions& options, in InstantiateRemote()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc54 const FunctionLibraryRuntime::InstantiateOptions& options, in ConstructFunctionGraphHelper()
64 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
74 const FunctionLibraryRuntime::InstantiateOptions& options, in InstantiateAndRun()
122 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
380 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
395 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
Dcluster_function_library_runtime.h40 const FunctionLibraryRuntime::InstantiateOptions& options,
57 const FunctionLibraryRuntime::InstantiateOptions& options,
Dcluster_function_library_runtime.cc38 const FunctionLibraryRuntime::InstantiateOptions& options, in ConstructFunctionGraph()
183 AttrSlice attrs, const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
/external/tensorflow/tensorflow/core/framework/
Dfunction.h532 struct InstantiateOptions { struct
643 const InstantiateOptions& options, argument
647 auto opts = absl::make_unique<InstantiateOptions>(); in Instantiate()
797 static string ExecutorType(const InstantiateOptions& options,
809 const FunctionLibraryRuntime::InstantiateOptions& options);
838 const FunctionLibraryRuntime::InstantiateOptions& options,
Dfunction_handle_cache.h35 FunctionLibraryRuntime::InstantiateOptions options,
Dfunction_handle_cache.cc38 FunctionLibraryRuntime::InstantiateOptions options, in Instantiate()
Dfunction.cc924 string FunctionLibraryRuntime::ExecutorType(const InstantiateOptions& options, in ExecutorType()
997 const FunctionLibraryRuntime::InstantiateOptions& options) { in Canonicalize()
1058 static const FunctionLibraryRuntime::InstantiateOptions* kEmptyOptions = in Canonicalize()
1059 new FunctionLibraryRuntime::InstantiateOptions; in Canonicalize()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.h46 FunctionLibraryRuntime::InstantiateOptions* opts);
Dpartitioned_function_ops.cc135 AttrSlice attrs, FunctionLibraryRuntime::InstantiateOptions* opts) { in FillOutputDevices()
166 FunctionLibraryRuntime::InstantiateOptions opts; in Instantiate()
Dfunction_ops.cc328 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in ComputeAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dcluster_function_library_runtime.h47 const FunctionLibraryRuntime::InstantiateOptions& options,
Dcluster_function_library_runtime.cc35 AttrSlice attrs, const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
Deager_service_impl_test.cc557 FunctionLibraryRuntime::InstantiateOptions options; in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc141 FunctionLibraryRuntime::InstantiateOptions options; in InstantiateFunc()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc244 FunctionLibraryRuntime::InstantiateOptions inst_ops; in ConstructFunctionHandle()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc533 FunctionLibraryRuntime::InstantiateOptions inst_opts; in Instantiate()