Home
last modified time | relevance | path

Searched refs:enable_automatic_model_parallelism (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/tpu/
H A Dcompile_metadata.proto117 bool enable_automatic_model_parallelism = 14; field
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.h136 bool enable_automatic_model_parallelism, bool enable_xla_param_broadcast,
H A Ddistributed_tpu_rewrite_pass.cc5074 bool enable_automatic_model_parallelism, bool enable_xla_param_broadcast, in SetDistributedTpuRewritePassOptions() argument
5082 enable_automatic_model_parallelism_ = enable_automatic_model_parallelism; in SetDistributedTpuRewritePassOptions()