Searched refs:backward_conv_shape (Results 1 – 1 of 1) sorted by relevance
287 Shape backward_conv_shape = backward_conv->shape().tuple_shapes(0); in CanonicalizeBackwardInputConvolution() local289 Shape new_backward_conv_shape = backward_conv_shape; in CanonicalizeBackwardInputConvolution()382 CHECK(ShapeUtil::Compatible(slice_shape, backward_conv_shape)) in CanonicalizeBackwardInputConvolution()384 << ShapeUtil::HumanString(backward_conv_shape); in CanonicalizeBackwardInputConvolution()387 HloInstruction::CreateSlice(backward_conv_shape, new_backward_conv, in CanonicalizeBackwardInputConvolution()