Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h97 Shape MakeNonPaddedShapeForGivenPartition(const Shape& shape,
Dspmd_partitioner_util.cc166 Shape MakeNonPaddedShapeForGivenPartition(const Shape& shape, in MakeNonPaddedShapeForGivenPartition() function
172 subshapes.push_back(MakeNonPaddedShapeForGivenPartition( in MakeNonPaddedShapeForGivenPartition()
Dspmd_partitioner.cc2525 MakeNonPaddedShapeForGivenPartition(shape, sharding, i); in HandleInfeed()
2927 MakeNonPaddedShapeForGivenPartition(shape, sharding, i); in HandleOutfeed()