Searched refs:weights_layout_ (Results 1 – 8 of 8) sorted by relevance
40 weights_layout_ = WeightsLayout::kOHWIOGroupO4I4; in ConvolutionTransposed()42 weights_layout_ = WeightsLayout::kOHWIOGroupI4O4; in ConvolutionTransposed()46 weights_layout_ = WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4; in ConvolutionTransposed()48 weights_layout_ = WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4; in ConvolutionTransposed()86 weights_layout_ = WeightsLayout::kOHWIOGroupO4I4; in ConvolutionTransposed()88 weights_layout_ = WeightsLayout::kOHWIOGroupI4O4; in ConvolutionTransposed()92 weights_layout_ = WeightsLayout::k2DX4O4YIsHWIAndXIsOOGroupI4; in ConvolutionTransposed()94 weights_layout_ = WeightsLayout::k2DX4I4YIsHWIAndXIsOOGroupO4; in ConvolutionTransposed()137 if (weights_layout_ == WeightsLayout::kOHWIOGroupI4O4 || in GenerateConvolutionTransposedCode()138 weights_layout_ == WeightsLayout::kOHWIOGroupO4I4) { in GenerateConvolutionTransposedCode()
53 desc.layout = weights_layout_; in GetWeightsDescription()79 WeightsLayout weights_layout_; variable
56 desc.layout = weights_layout_; in GetWeightsDescription()90 WeightsLayout weights_layout_; variable
58 desc.layout = weights_layout_; in GetWeightsDescription()91 WeightsLayout weights_layout_; variable
57 desc.layout = weights_layout_; in GetWeightsDescription()94 WeightsLayout weights_layout_; variable
32 weights_layout_ = WeightsLayout::kOICustomSpatialO4I4; in ConvolutionTransposed3x3Thin()34 weights_layout_ = WeightsLayout::kOICustomSpatialI4O4; in ConvolutionTransposed3x3Thin()
58 weights_layout_ = WeightsLayout::kOICustomSpatialO4I4; in ConvolutionTransposed4x4()60 weights_layout_ = WeightsLayout::kOICustomSpatialI4O4; in ConvolutionTransposed4x4()
42 weights_layout_ = WeightsLayout::kOICustomSpatialO4I4; in ConvolutionTransposed3x3()44 weights_layout_ = WeightsLayout::kOICustomSpatialI4O4; in ConvolutionTransposed3x3()