Searched refs:convolution_shape (Results 1 – 2 of 2) sorted by relevance
883 const Shape& convolution_shape = convolution->shape(); in HandleConvolution() local887 LayoutUtil::IsMonotonicWithDim0Major(convolution_shape.layout())) { in HandleConvolution()926 const Shape& convolution_shape = convolution->shape(); in HandleConvolution() local928 convolution_shape.dimensions(dnums.output_spatial_dimensions(0)); in HandleConvolution()931 : convolution_shape.dimensions( in HandleConvolution()
5419 const Shape& convolution_shape = convolution->shape(); in SimplifyConvToDot() local5422 TF_RET_CHECK(LayoutUtil::HasLayout(convolution_shape)); in SimplifyConvToDot()5453 if (!LayoutUtil::Equal(input_shape.layout(), convolution_shape.layout()) || in SimplifyConvToDot()5456 LayoutUtil::Minor(convolution_shape.layout(), 0) != in SimplifyConvToDot()5500 convolution_shape.element_type(), {conv_width, output_channels}); in SimplifyConvToDot()5513 ReplaceInstruction(convolution, add_bitcast(convolution_shape, dot))); in SimplifyConvToDot()