Searched refs:HandleDotHelper (Results 1 – 3 of 3) sorted by relevance
442 Status HandleDotHelper(HloInstruction* hlo,
1043 return HandleDotHelper(hlo, mapping, create_sharded_conv); in HandleConvolution()
90 return HandleDotHelper(hlo, mapping, create_sharded_dot); in HandleDot()2835 Status SpmdPartitioningVisitor::HandleDotHelper( in HandleDotHelper() function in xla::spmd::SpmdPartitioningVisitor