Searched defs:conv_dims (Results 1 – 3 of 3) sorted by relevance
55 const auto& conv_dims = conv->convolution_dimension_numbers(); in ParseConvolutionDimsInfo() local
160 std::vector<int64> conv_dims = in Compile() local
979 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in ConvolutionDimensionsFromInputs() local1005 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DOperations() local1448 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DBackpropInputOperations() local1505 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DBackpropFilterOperations() local2165 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in OpDimensionsFromInputs() local