Searched refs:original_dot (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 39 Status CanonicalizeDot(HloInstruction* original_dot) { in CanonicalizeDot() argument
40 auto computation = original_dot->parent(); in CanonicalizeDot()
41 const auto& original_dnums = original_dot->dot_dimension_numbers(); in CanonicalizeDot()
46 const auto& lhs_shape = original_dot->operand(0)->shape(); in CanonicalizeDot()
84 original_dot->mutable_operand(0), lhs_transpose)); in CanonicalizeDot()
96 const auto& rhs_shape = original_dot->operand(1)->shape(); in CanonicalizeDot()
130 original_dot->mutable_operand(1), rhs_transpose)); in CanonicalizeDot()
161 ShapeUtil::MakeShape(original_dot->shape().element_type(), dot_dims), in CanonicalizeDot()
162 reshaped_lhs, reshaped_rhs, dot_dnums, original_dot->precision_config())); in CanonicalizeDot()
165 original_dot, computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 137 Window GenNewWindow(const HloInstruction* original_dot, in GenNewWindow() argument
142 auto new_window = original_dot->window(); in GenNewWindow()
144 original_dot->convolution_dimension_numbers(); in GenNewWindow()
222 const HloInstruction* original_dot, const HloInstruction* dot_lhs, in GenNewConvDNums() argument
230 original_dot->convolution_dimension_numbers(); in GenNewConvDNums()
931 TF_ASSIGN_OR_RETURN(auto original_dot, in PartitionBaseCase()
934 VLOG(2) << original_dot->ToString(); in PartitionBaseCase()
937 auto original_sharded_dot_shape = original_dot->shape(); in PartitionBaseCase()
981 original_dot->feature_group_count(), in PartitionBaseCase()
982 original_dot->batch_group_count(), in PartitionBaseCase()
[all …]
|