Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2090 int64 merge_factor = output_dim_size / input_dim_size; in HandleReshape() local
2095 input_shard_shape.dimensions(input_sharded_dim) * merge_factor); in HandleReshape()