Home
last modified time | relevance | path

Searched defs:padded_base_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc285 Shape padded_base_shape = base_shape; in GetPaddedShapeForUnevenPartitioning() local
295 auto padded_base_shape = in PadBaseShapeBeforeUnevenTiledSharding() local
Dspmd_partitioner.cc1027 auto padded_base_shape = shard_shape; in ReshardFromPartialReplicateWithDynamicSlice() local