Searched defs:gather_shape (Results 1 – 1 of 1) sorted by relevance
903 const Shape gather_shape = gather.shape(); in GatherParallelDataOperandSharding() local1341 const Shape gather_shape = gather.shape(); in GatherParallelOutputDims() local