Searched defs:conv_dims (Results 1 – 3 of 3) sorted by relevance
160 std::vector<int64> conv_dims = in Compile() local
81 const auto& conv_dims = conv->convolution_dimension_numbers(); in ParseConvolutionDimsInfo() local
990 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in ConvolutionDimensionsFromInputs() local1016 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DOperations() local1459 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DBackpropInputOperations() local1516 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DBackpropFilterOperations() local2178 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in OpDimensionsFromInputs() local