Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2019 int64 input_major_dims_size = 1; in HandleReshape() local
2021 input_major_dims_size *= operand.base_shape().dimensions(i); in HandleReshape()
2027 if (input_major_dims_size != output_major_dims_size) { in HandleReshape()