Home
last modified time | relevance | path

Searched refs:is_multi_device_function (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dpartitioned_function_ops.cc205 opts.is_multi_device_function = true; in Instantiate()
H A Dbatch_kernels.cc372 opts.is_multi_device_function = true; in InstantiateFunction()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dcaptured_function.cc554 inst_opts.is_multi_device_function = metadata_->use_multi_device_function(); in Instantiate()
567 if (inst_opts.is_multi_device_function) { in Instantiate()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device.cc165 options.is_multi_device_function = true; in InstantiateFunc()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dprocess_function_library_runtime_test.cc504 instantiate_opts.is_multi_device_function = true; in TEST_F()
693 inst_opts.is_multi_device_function = true; in MakeOptions()
H A Dfunction.cc729 if (options.is_multi_device_function) return false; in IsLocalTarget()
H A Dprocess_function_library_runtime.cc1469 if (options.is_multi_device_function) { in Instantiate()
H A Dfunction_test.cc416 options.is_multi_device_function = true; in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfunction.h649 bool is_multi_device_function = false; member
/aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/
H A Drpc_ops.cc659 instantiate_opts.is_multi_device_function = true; in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Deager_service_impl_test.cc847 options.is_multi_device_function = true; in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
H A Dtrt_engine_op.cc371 inst_ops.is_multi_device_function = true; in ConstructFunctionHandle()