Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2023 int64 output_major_dims_size = 1; in HandleReshape() local
2025 output_major_dims_size *= hlo->shape().dimensions(i); in HandleReshape()
2027 if (input_major_dims_size != output_major_dims_size) { in HandleReshape()