Searched refs:PartitionDot (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 1636 StatusOr<HloInstruction*> PartitionDot(
1868 PartitionDot(per_group_lhs, per_group_rhs, in PartitionDotGroupOnBatch()
2065 PartitionDot(lhs_matching ? matching_p : other_p, in PartitionDotGroupOnNonContracting()
2224 PartitionDot( in PartitionDotGroupOnContracting()
2431 StatusOr<HloInstruction*> PartitionDot( in PartitionDot() function
2767 PartitionDot(PartitionedHlo(lhs.hlo(), lhs.base_shape(), inner_state), in PartitionDot()
2798 StatusOr<HloInstruction*> PartitionDot( in PartitionDot() function
2814 PartitionDot(lhs, rhs, output_base_shape, output_sharding, dims_mapping, in PartitionDot()
2849 PartitionDot(lhs, rhs, hlo->shape(), hlo->sharding(), dims_mapping, in HandleDotHelper()
|