Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_util.h154 Shape MakeNonPaddedShapeForGivenPartition(const Shape& shape,
H A Dspmd_partitioner_util.cc115 Shape MakeNonPaddedShapeForGivenPartition(const Shape& shape, in MakeNonPaddedShapeForGivenPartition() function
123 subshapes.push_back(MakeNonPaddedShapeForGivenPartition( in MakeNonPaddedShapeForGivenPartition()
H A Dspmd_partitioner.cc3280 MakeNonPaddedShapeForGivenPartition(shape, sharding, i); in HandleInfeed()
3708 MakeNonPaddedShapeForGivenPartition(shape, sharding, i); in HandleOutfeed()