Searched refs:backward_conv (Results 1 – 2 of 2) sorted by relevance
203 HloInstruction* backward_conv) { in CanonicalizeBackwardFilterConvolution() argument204 CHECK_EQ(backward_conv->custom_call_target(), in CanonicalizeBackwardFilterConvolution()206 if (window_util::HasSymmetricPadding(backward_conv->window())) { in CanonicalizeBackwardFilterConvolution()218 HloInstruction* input = backward_conv->mutable_operand(0); in CanonicalizeBackwardFilterConvolution()219 Window new_backward_conv_window = backward_conv->window(); in CanonicalizeBackwardFilterConvolution()224 backward_conv->convolution_dimension_numbers(); in CanonicalizeBackwardFilterConvolution()225 for (size_t i = 0; i < backward_conv->window().dimensions_size(); ++i) { in CanonicalizeBackwardFilterConvolution()226 int64 padding_low = backward_conv->window().dimensions(i).padding_low(); in CanonicalizeBackwardFilterConvolution()227 int64 padding_high = backward_conv->window().dimensions(i).padding_high(); in CanonicalizeBackwardFilterConvolution()251 HloComputation* computation = backward_conv->parent(); in CanonicalizeBackwardFilterConvolution()[all …]
39 bool CanonicalizeBackwardFilterConvolution(HloInstruction* backward_conv);40 bool CanonicalizeBackwardInputConvolution(HloInstruction* backward_conv);