Home
last modified time | relevance | path

Searched refs:set_allow_spmd_sharding_propagation_to_output (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.cc126 output.set_allow_spmd_sharding_propagation_to_output( in ToProto()
157 output.set_allow_spmd_sharding_propagation_to_output( in ExecutableBuildOptionsFromProto()
208 execution_options.set_allow_spmd_sharding_propagation_to_output( in CreateExecutionOptions()
H A Dexecutable_build_options.h156 ExecutableBuildOptions& set_allow_spmd_sharding_propagation_to_output( in set_allow_spmd_sharding_propagation_to_output() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_util.cc98 config->set_allow_spmd_sharding_propagation_to_output( in CreateModuleConfig()
H A Dhlo_module_config.h312 void set_allow_spmd_sharding_propagation_to_output( in set_allow_spmd_sharding_propagation_to_output() function
H A Dhlo_module.cc551 module_config.set_allow_spmd_sharding_propagation_to_output( in CreateModuleConfigFromShape()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/c/
H A Dpjrt_c_api_wrapper_impl.cc213 ret.executable_build_options.set_allow_spmd_sharding_propagation_to_output( in ConvertCCompileOptionstoCppCompileOptions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla_compiler.cc735 set_allow_spmd_sharding_propagation_to_output); in BuildXlaCompilerSubmodule()