Searched refs:conv_weights_desc (Results 1 – 2 of 2) sorted by relevance
48 const OperationDef& op_def, const WeightsDescription& conv_weights_desc) { in GetConverterToConvWeightsCode() argument55 if (conv_weights_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetConverterToConvWeightsCode()56 conv_weights_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetConverterToConvWeightsCode()57 std::vector<int32_t> remap(conv_weights_desc.spatial_remap.size()); in GetConverterToConvWeightsCode()59 remap[i] = conv_weights_desc.spatial_remap[i]; in GetConverterToConvWeightsCode()83 if (conv_weights_desc.layout == WeightsLayout::kOICustomSpatialI4O4 || in GetConverterToConvWeightsCode()84 conv_weights_desc.layout == WeightsLayout::kOICustomSpatialO4I4) { in GetConverterToConvWeightsCode()114 if (conv_weights_desc.IsI4O4()) { in GetConverterToConvWeightsCode()119 } else if (conv_weights_desc.IsO4I4()) { in GetConverterToConvWeightsCode()125 if (conv_weights_desc.layout == WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4 || in GetConverterToConvWeightsCode()[all …]
227 WeightsDescription conv_weights_desc; in GPUOperationFromNode() local230 &conv_weights_desc); in GPUOperationFromNode()233 AlignByN(weights_shape.b, conv_weights_desc.GetOutputGroupSize() * 4); in GPUOperationFromNode()248 SelectConverterToConvWeights(conv_weights_desc, converter_def, hints); in GPUOperationFromNode()304 WeightsDescription conv_weights_desc; in GPUOperationFromNode() local307 &conv_weights_desc); in GPUOperationFromNode()310 weights_shape.b, conv_weights_desc.GetOutputGroupSize() * 4); in GPUOperationFromNode()325 conv_weights_desc, converter_def, hints); in GPUOperationFromNode()