afe.backends.mla.mla_checkers ============================= .. py:module:: afe.backends.mla.mla_checkers .. autoapi-nested-parse:: This file contains the MLAChecker class and the check functions for all MLA supported IRs. Attributes ---------- .. autoapisummary:: afe.backends.mla.mla_checkers.FORCE_SUPPORT_ALIGN_CORNER_TRUE afe.backends.mla.mla_checkers.baseline_test afe.backends.mla.mla_checkers.pool_test afe.backends.mla.mla_checkers.max_pool_test afe.backends.mla.mla_checkers.avg_pool_test afe.backends.mla.mla_checkers.binary_elementwise_test afe.backends.mla.mla_checkers.reduce_test afe.backends.mla.mla_checkers.conv_transpose_test afe.backends.mla.mla_checkers.group_conv2d_transpose_test afe.backends.mla.mla_checkers.upsampling_test afe.backends.mla.mla_checkers.image_resize_test afe.backends.mla.mla_checkers.tuple_concat_test afe.backends.mla.mla_checkers.strided_slice_test afe.backends.mla.mla_checkers.argmax_test afe.backends.mla.mla_checkers.softmax_test afe.backends.mla.mla_checkers.variance_test afe.backends.mla.mla_checkers.gridsample_test afe.backends.mla.mla_checkers.conv_test afe.backends.mla.mla_checkers.broadcast_to_test afe.backends.mla.mla_checkers.transpose_test afe.backends.mla.mla_checkers.MLA_REDUCE_OPS afe.backends.mla.mla_checkers.MLA_ADAPTIVE_POOL_OPS afe.backends.mla.mla_checkers.MLA_ELEMENTWISE_OPS afe.backends.mla.mla_checkers.MLA_BASELINE_CHECKS_ONLY_OPS afe.backends.mla.mla_checkers.MLA_CONV_OPS afe.backends.mla.mla_checkers.MLA_CONV_TRANSPOSE_OPS afe.backends.mla.mla_checkers.MLA_EXEMPT_FROM_BASELINE_CHECKS afe.backends.mla.mla_checkers.MLA_5D_OPS Classes ------- .. autoapisummary:: afe.backends.mla.mla_checkers.MLAChecker Functions --------- .. autoapisummary:: afe.backends.mla.mla_checkers.tuple_einsum_test Module Contents --------------- .. py:data:: FORCE_SUPPORT_ALIGN_CORNER_TRUE :value: True .. py:data:: baseline_test .. py:data:: pool_test .. py:data:: max_pool_test .. py:data:: avg_pool_test .. py:data:: binary_elementwise_test .. py:data:: reduce_test .. py:data:: conv_transpose_test .. py:data:: group_conv2d_transpose_test .. py:data:: upsampling_test .. py:data:: image_resize_test .. py:data:: tuple_concat_test .. py:data:: strided_slice_test .. py:data:: argmax_test .. py:data:: softmax_test .. py:data:: variance_test .. py:data:: gridsample_test .. py:function:: tuple_einsum_test(num_inputs: int) -> afe.backends.backend_checker.Predicate .. py:data:: conv_test .. py:data:: broadcast_to_test .. py:data:: transpose_test .. py:data:: MLA_REDUCE_OPS .. py:data:: MLA_ADAPTIVE_POOL_OPS .. py:data:: MLA_ELEMENTWISE_OPS .. py:data:: MLA_BASELINE_CHECKS_ONLY_OPS .. py:data:: MLA_CONV_OPS :value: ['conv2d', 'conv2d_clip', 'conv2d_relu', 'conv2d_add', 'conv2d_add_clip', 'conv2d_add_relu',... .. py:data:: MLA_CONV_TRANSPOSE_OPS :value: ['conv2dtranspose', 'conv2dtranspose_clip', 'conv2dtranspose_relu', 'conv2dtranspose_add',... .. py:data:: MLA_EXEMPT_FROM_BASELINE_CHECKS .. py:data:: MLA_5D_OPS .. py:class:: MLAChecker Checker class for MLA backend. See BaseChecker for detailed documentation.