Searched refs:conv_dims (Results 1 – 3 of 3) sorted by relevance
632 OpLevelCostEstimator::ConvolutionDimensions conv_dims = { in ConvolutionDimensionsFromInputs() local643 return conv_dims; in ConvolutionDimensionsFromInputs()652 ConvolutionDimensions conv_dims = ConvolutionDimensionsFromInputs( in CountConv2DOperations() local661 int64 ops = conv_dims.batch; in CountConv2DOperations()662 ops *= conv_dims.ox * conv_dims.oy; in CountConv2DOperations()663 ops *= conv_dims.kx * conv_dims.ky; in CountConv2DOperations()665 ops *= conv_dims.iz * conv_dims.oz; in CountConv2DOperations()669 conv_dims.oz *= conv_dims.iz; in CountConv2DOperations()670 ops *= conv_dims.oz; in CountConv2DOperations()675 *conv_info = conv_dims; in CountConv2DOperations()[all …]
937 conv_dims=None): argument1024 if conv_dims is not None and conv_dims + 2 != input_rank:1026 (conv_dims + 2, input_rank))1112 conv_dims=1)1155 conv_dims=2)1198 conv_dims=3)
174 conv_dims = input_shape.ndims - 2176 strides = [1] * conv_dims177 elif len(strides) != conv_dims:179 conv_dims))180 if conv_dims == 1:189 elif conv_dims == 2:200 elif conv_dims == 3: