Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2045 auto input_shard_shape = in HandleReshape() local
2095 input_shard_shape.dimensions(input_sharded_dim) * merge_factor); in HandleReshape()