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