afe.backends.mla.mla_checkers

This file contains the MLAChecker class and the check functions for all MLA supported IRs.

Classes

Checker class for MLA backend. See BaseChecker for detailed documentation.

tuple_einsum_test(→ afe.backends.backend_checker.Predicate)

afe.backends.mla.mla_checkers.FORCE_SUPPORT_ALIGN_CORNER_TRUE = True[source]

afe.backends.mla.mla_checkers.tuple_einsum_test(num_inputs: int) → afe.backends.backend_checker.Predicate[source]

afe.backends.mla.mla_checkers.MLA_CONV_OPS = ['conv2d', 'conv2d_clip', 'conv2d_relu', 'conv2d_add', 'conv2d_add_clip', 'conv2d_add_relu',...[source]

afe.backends.mla.mla_checkers.MLA_CONV_TRANSPOSE_OPS = ['conv2dtranspose', 'conv2dtranspose_clip', 'conv2dtranspose_relu', 'conv2dtranspose_add',...[source]

class afe.backends.mla.mla_checkers.MLAChecker[source]: Checker class for MLA backend. See BaseChecker for detailed documentation.

`FORCE_SUPPORT_ALIGN_CORNER_TRUE`
`baseline_test`
`pool_test`
`max_pool_test`
`avg_pool_test`
`binary_elementwise_test`
`reduce_test`
`conv_transpose_test`
`group_conv2d_transpose_test`
`upsampling_test`
`image_resize_test`
`tuple_concat_test`
`strided_slice_test`
`argmax_test`
`softmax_test`
`variance_test`
`gridsample_test`
`conv_test`
`broadcast_to_test`
`transpose_test`
`MLA_REDUCE_OPS`
`MLA_ADAPTIVE_POOL_OPS`
`MLA_ELEMENTWISE_OPS`
`MLA_BASELINE_CHECKS_ONLY_OPS`
`MLA_CONV_OPS`
`MLA_CONV_TRANSPOSE_OPS`
`MLA_EXEMPT_FROM_BASELINE_CHECKS`
`MLA_5D_OPS`