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