Searched refs:ConvolutionType (Results 1 – 2 of 2) sorted by relevance
60 case ConvolutionType::kConv: { in Build()72 case ConvolutionType::kDepthwiseConv: { in Build()86 case ConvolutionType::kTransposeConv: { in Build()134 if (conv_type_ == ConvolutionType::kDepthwiseConv) { in FillCoreMLWeights()158 if (conv_type_ == ConvolutionType::kDepthwiseConv) { in TransposeKernelWeights()168 if (conv_type_ == ConvolutionType::kTransposeConv) { in TransposeKernelWeights()214 case ConvolutionType::kConv: { in PopulateSubgraph()220 case ConvolutionType::kDepthwiseConv: { in PopulateSubgraph()226 case ConvolutionType::kTransposeConv: { in PopulateSubgraph()248 if (conv_type_ == ConvolutionType::kTransposeConv) { in RegisterInputs()[all …]
29 enum class ConvolutionType { kConv, kDepthwiseConv, kTransposeConv }; enum35 ConvolutionType conv_type) in ConvolutionOpBuilder()73 ConvolutionType conv_type_;