Searched defs:conv_dims (Results 1 – 3 of 3) sorted by relevance
160 std::vector<int64_t> conv_dims = in Compile() local
82 const auto& conv_dims = conv->convolution_dimension_numbers(); in ParseConvolutionDimsInfo() local
991 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in ConvolutionDimensionsFromInputs() local1017 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DOperations() local1460 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DBackpropInputOperations() local1517 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DBackpropFilterOperations() local2204 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in OpDimensionsFromInputs() local