Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2050 int64 split_factor = input_dim_size / output_dim_size; in HandleReshape() local
2063 dim->set_padding_high(output_shard_size * split_factor * in HandleReshape()
2081 output_shard_size * split_factor); in HandleReshape()