Searched refs:convolution_shape (Results 1 – 2 of 2) sorted by relevance
3546 const Shape& convolution_shape = convolution->shape(); in SimplifyConvToDot() local3549 TF_RET_CHECK(LayoutUtil::HasLayout(convolution_shape)); in SimplifyConvToDot()3580 if (!LayoutUtil::Equal(input_shape.layout(), convolution_shape.layout()) || in SimplifyConvToDot()3583 LayoutUtil::Minor(convolution_shape.layout(), 0) != in SimplifyConvToDot()3625 convolution_shape.element_type(), {conv_width, output_channels}); in SimplifyConvToDot()3637 ReplaceInstruction(convolution, add_bitcast(convolution_shape, dot))); in SimplifyConvToDot()
1106 const Shape& convolution_shape = convolution->shape(); in HandleConvolution() local1110 LayoutUtil::IsMonotonicWithDim0Major(convolution_shape.layout())) { in HandleConvolution()1149 const Shape& convolution_shape = convolution->shape(); in HandleConvolution() local1151 convolution_shape.dimensions(dnums.output_spatial_dimensions(0)); in HandleConvolution()1154 : convolution_shape.dimensions( in HandleConvolution()