Home
last modified time | relevance | path

Searched refs:create_sharded_dot (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc78 auto create_sharded_dot = in HandleDot() local
90 return HandleDotHelper(hlo, mapping, create_sharded_dot); in HandleDot()
472 const Window& conv_window)>& create_sharded_dot, in PartitionBaseCase() argument
525 auto dot, create_sharded_dot(lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionBaseCase()
552 create_sharded_dot(lhs.hlo(), resharded_rhs.hlo(), b, conv_window)); in PartitionBaseCase()
569 create_sharded_dot(resharded_lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionBaseCase()
932 create_sharded_dot(original_dot_lhs, original_dot_rhs, in PartitionBaseCase()
1105 auto dot, create_sharded_dot(dot_lhs, dot_rhs, &body_b, conv_window)); in PartitionBaseCase()
1512 auto dot, create_sharded_dot(lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionBaseCase()
1533 TF_ASSIGN_OR_RETURN(auto dot, create_sharded_dot(lhs.hlo(), rhs_replicated, in PartitionBaseCase()
[all …]
Dspmd_partitioner.h446 const Window& conv_window)>& create_sharded_dot);