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()336 HloInstruction* new_backward_conv = in CanonicalizeBackwardInputConvolution() local347 std::vector<int64> start_indices(new_backward_conv->shape().dimensions_size(), in CanonicalizeBackwardInputConvolution()350 new_backward_conv->shape().dimensions().begin(), in CanonicalizeBackwardInputConvolution()351 new_backward_conv->shape().dimensions().end()); in CanonicalizeBackwardInputConvolution()352 std::vector<int64> strides(new_backward_conv->shape().dimensions_size(), 1LL); in CanonicalizeBackwardInputConvolution()371 ShapeInference::InferSliceShape(new_backward_conv->shape(), start_indices, in CanonicalizeBackwardInputConvolution()[all …]