Searched refs:new_backward_conv (Results 1 – 1 of 1) sorted by relevance
261 HloInstruction* new_backward_conv = in CanonicalizeBackwardFilterConvolution() local264 new_backward_conv->set_window(new_backward_conv_window); in CanonicalizeBackwardFilterConvolution()268 << new_backward_conv->ToString(); in CanonicalizeBackwardFilterConvolution()271 computation->ReplaceInstruction(backward_conv, new_backward_conv)); in CanonicalizeBackwardFilterConvolution()340 HloInstruction* new_backward_conv = in CanonicalizeBackwardInputConvolution() local351 std::vector<int64> start_indices(new_backward_conv->shape().dimensions_size(), in CanonicalizeBackwardInputConvolution()354 new_backward_conv->shape().dimensions().begin(), in CanonicalizeBackwardInputConvolution()355 new_backward_conv->shape().dimensions().end()); in CanonicalizeBackwardInputConvolution()356 std::vector<int64> strides(new_backward_conv->shape().dimensions_size(), 1LL); in CanonicalizeBackwardInputConvolution()379 ShapeInference::InferSliceShape(new_backward_conv->shape(), start_indices, in CanonicalizeBackwardInputConvolution()[all …]