24.02.1
|
Functions | |
arm_gemm::Activation | map_to_arm_gemm_activation (const ActivationLayerInfo &act) |
Performs a mapping between Compute Library ActivationLayerInfo and the assembly Activation structure. More... | |
arm_conv::PaddingValues | map_to_arm_conv_padding (const PadStrideInfo &pad_stride_info) |
Performs a mapping between Compute Library PadStrideInfo and the assembly PaddingValues structure. More... | |
arm_gemm::WeightFormat | map_to_arm_gemm_weight_format (const arm_compute::WeightFormat &weight_format) |
Performs a mapping from Compute Library WeightFormat to the assembly WeightFormat enum. More... | |
arm_compute::WeightFormat | map_to_arm_compute_weight_format (const arm_gemm::WeightFormat &weight_format) |
Performs a mapping from Assembly WeightFormat to the Compute Library WeightFormat enum. More... | |
arm_compute::WeightFormat map_to_arm_compute_weight_format | ( | const arm_gemm::WeightFormat & | weight_format | ) |
Performs a mapping from Assembly WeightFormat to the Compute Library WeightFormat enum.
[in] | weight_format | Assembly WeightFormat enum value |
Definition at line 190 of file AssemblyUtils.cpp.
References arm_gemm::ANY, arm_compute::ANY, arm_gemm::OHWI, arm_compute::OHWI, arm_gemm::OHWIo128, arm_compute::OHWIo128, arm_gemm::OHWIo16, arm_compute::OHWIo16, arm_gemm::OHWIo16i2, arm_compute::OHWIo16i2, arm_gemm::OHWIo16i2_bf16, arm_compute::OHWIo16i2_bf16, arm_gemm::OHWIo16i4, arm_compute::OHWIo16i4, arm_gemm::OHWIo16i4_bf16, arm_compute::OHWIo16i4_bf16, arm_gemm::OHWIo16i8, arm_compute::OHWIo16i8, arm_gemm::OHWIo2, arm_compute::OHWIo2, arm_gemm::OHWIo2i8, arm_compute::OHWIo2i8, arm_gemm::OHWIo32, arm_compute::OHWIo32, arm_gemm::OHWIo32i2, arm_compute::OHWIo32i2, arm_gemm::OHWIo32i2_bf16, arm_compute::OHWIo32i2_bf16, arm_gemm::OHWIo32i4, arm_compute::OHWIo32i4, arm_gemm::OHWIo32i4_bf16, arm_compute::OHWIo32i4_bf16, arm_gemm::OHWIo32i8, arm_compute::OHWIo32i8, arm_gemm::OHWIo4, arm_compute::OHWIo4, arm_gemm::OHWIo4i2, arm_compute::OHWIo4i2, arm_gemm::OHWIo4i2_bf16, arm_compute::OHWIo4i2_bf16, arm_gemm::OHWIo4i4, arm_compute::OHWIo4i4, arm_gemm::OHWIo4i4_bf16, arm_compute::OHWIo4i4_bf16, arm_gemm::OHWIo4i8, arm_compute::OHWIo4i8, arm_gemm::OHWIo64, arm_compute::OHWIo64, arm_gemm::OHWIo64i2, arm_compute::OHWIo64i2, arm_gemm::OHWIo64i2_bf16, arm_compute::OHWIo64i2_bf16, arm_gemm::OHWIo64i4, arm_compute::OHWIo64i4, arm_gemm::OHWIo64i4_bf16, arm_compute::OHWIo64i4_bf16, arm_gemm::OHWIo64i8, arm_compute::OHWIo64i8, arm_gemm::OHWIo8, arm_compute::OHWIo8, arm_gemm::OHWIo8i2, arm_compute::OHWIo8i2, arm_gemm::OHWIo8i2_bf16, arm_compute::OHWIo8i2_bf16, arm_gemm::OHWIo8i4, arm_compute::OHWIo8i4, arm_gemm::OHWIo8i4_bf16, arm_compute::OHWIo8i4_bf16, arm_gemm::OHWIo8i8, arm_compute::OHWIo8i8, arm_gemm::UNSPECIFIED, and arm_compute::UNSPECIFIED.
Referenced by CpuGemmAssemblyDispatch::has_opt_impl().
arm_conv::PaddingValues map_to_arm_conv_padding | ( | const PadStrideInfo & | pad_stride_info | ) |
Performs a mapping between Compute Library PadStrideInfo and the assembly PaddingValues structure.
[in] | pad_stride_info | Compute Library padding and strides info. |
Definition at line 64 of file AssemblyUtils.cpp.
References PadStrideInfo::pad_bottom(), PadStrideInfo::pad_left(), PadStrideInfo::pad_right(), and PadStrideInfo::pad_top().
arm_gemm::Activation map_to_arm_gemm_activation | ( | const ActivationLayerInfo & | act | ) |
Performs a mapping between Compute Library ActivationLayerInfo and the assembly Activation structure.
[in] | act | Compute Library activation info. |
Definition at line 32 of file AssemblyUtils.cpp.
References ActivationLayerInfo::a(), ActivationLayerInfo::activation(), ActivationLayerInfo::b(), Activation::BoundedReLU, Activation::None, Activation::param1, Activation::param2, Activation::ReLU, and Activation::type.
Referenced by CpuGemmAssemblyDispatch::configure(), CpuGemmAssemblyDispatch::has_opt_impl(), CpuDepthwiseConv2dAssemblyDispatch::is_activation_supported(), and CpuGemmAssemblyDispatch::is_activation_supported().
arm_gemm::WeightFormat map_to_arm_gemm_weight_format | ( | const arm_compute::WeightFormat & | weight_format | ) |
Performs a mapping from Compute Library WeightFormat to the assembly WeightFormat enum.
[in] | weight_format | Compute Library WeightFormat enum value |
Definition at line 70 of file AssemblyUtils.cpp.
References arm_gemm::ANY, arm_compute::ANY, arm_gemm::OHWI, arm_compute::OHWI, arm_gemm::OHWIo128, arm_compute::OHWIo128, arm_gemm::OHWIo16, arm_compute::OHWIo16, arm_gemm::OHWIo16i2, arm_compute::OHWIo16i2, arm_gemm::OHWIo16i2_bf16, arm_compute::OHWIo16i2_bf16, arm_gemm::OHWIo16i4, arm_compute::OHWIo16i4, arm_gemm::OHWIo16i4_bf16, arm_compute::OHWIo16i4_bf16, arm_gemm::OHWIo16i8, arm_compute::OHWIo16i8, arm_gemm::OHWIo2, arm_compute::OHWIo2, arm_gemm::OHWIo2i8, arm_compute::OHWIo2i8, arm_gemm::OHWIo32, arm_compute::OHWIo32, arm_gemm::OHWIo32i2, arm_compute::OHWIo32i2, arm_gemm::OHWIo32i2_bf16, arm_compute::OHWIo32i2_bf16, arm_gemm::OHWIo32i4, arm_compute::OHWIo32i4, arm_gemm::OHWIo32i4_bf16, arm_compute::OHWIo32i4_bf16, arm_gemm::OHWIo32i8, arm_compute::OHWIo32i8, arm_gemm::OHWIo4, arm_compute::OHWIo4, arm_gemm::OHWIo4i2, arm_compute::OHWIo4i2, arm_gemm::OHWIo4i2_bf16, arm_compute::OHWIo4i2_bf16, arm_gemm::OHWIo4i4, arm_compute::OHWIo4i4, arm_gemm::OHWIo4i4_bf16, arm_compute::OHWIo4i4_bf16, arm_gemm::OHWIo4i8, arm_compute::OHWIo4i8, arm_gemm::OHWIo64, arm_compute::OHWIo64, arm_gemm::OHWIo64i2, arm_compute::OHWIo64i2, arm_gemm::OHWIo64i2_bf16, arm_compute::OHWIo64i2_bf16, arm_gemm::OHWIo64i4, arm_compute::OHWIo64i4, arm_gemm::OHWIo64i4_bf16, arm_compute::OHWIo64i4_bf16, arm_gemm::OHWIo64i8, arm_compute::OHWIo64i8, arm_gemm::OHWIo8, arm_compute::OHWIo8, arm_gemm::OHWIo8i2, arm_compute::OHWIo8i2, arm_gemm::OHWIo8i2_bf16, arm_compute::OHWIo8i2_bf16, arm_gemm::OHWIo8i4, arm_compute::OHWIo8i4, arm_gemm::OHWIo8i4_bf16, arm_compute::OHWIo8i4_bf16, arm_gemm::OHWIo8i8, arm_compute::OHWIo8i8, arm_gemm::UNSPECIFIED, and arm_compute::UNSPECIFIED.
Referenced by CpuGemmAssemblyDispatch::has_opt_impl().