Searched refs:convolution_shape (Results 1 – 2 of 2) sorted by relevance
919 const Shape& convolution_shape = convolution->shape(); in HandleConvolution() local923 LayoutUtil::IsMonotonicWithDim0Major(convolution_shape.layout())) { in HandleConvolution()966 const Shape& convolution_shape = convolution->shape(); in HandleConvolution() local968 output_dims.push_back(convolution_shape.dimensions(d)); in HandleConvolution()
6562 const Shape& convolution_shape = convolution->shape(); in SimplifyConvToDot() local6565 TF_RET_CHECK(LayoutUtil::HasLayout(convolution_shape)); in SimplifyConvToDot()6596 if (!LayoutUtil::Equal(input_shape.layout(), convolution_shape.layout()) || in SimplifyConvToDot()6599 LayoutUtil::Minor(convolution_shape.layout(), 0) != in SimplifyConvToDot()6647 convolution_shape.element_type(), {conv_width, output_channels}); in SimplifyConvToDot()6660 ReplaceInstruction(convolution, add_bitcast(convolution_shape, dot))); in SimplifyConvToDot()