afe.backends.mla.mla_checkers

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

Attributes

FORCE_SUPPORT_ALIGN_CORNER_TRUE
baseline_test
pool_test
max_pool_test
avg_pool_test
binary_elementwise_test
reduce_base_test
reduce_mean_test
reduce_max_test
conv_transpose_test
group_conv2d_transpose_test
upsampling_test
image_resize_test
depth_to_space_test
tuple_concat_shape_test
tuple_concat_test
strided_slice_test
argmax_test
softmax_test
variance_test
lrn_test
gridsample_test
conv_test
broadcast_to_test
transpose_test
reshape_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

Classes

MLAChecker	Checker class for MLA backend. See BaseChecker for detailed documentation.
RestrictedDimensionMLAChecker	Checker class for MLA backend, like MLAChecker except that it does not allow reshape on MLA.

Functions

tuple_einsum_test(→ afe.backends.backend_checker.Predicate)

Module Contents

afe.backends.mla.mla_checkers.FORCE_SUPPORT_ALIGN_CORNER_TRUE = 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_base_test

afe.backends.mla.mla_checkers.reduce_mean_test

afe.backends.mla.mla_checkers.reduce_max_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.depth_to_space_test

afe.backends.mla.mla_checkers.tuple_concat_shape_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.lrn_test

afe.backends.mla.mla_checkers.gridsample_test

afe.backends.mla.mla_checkers.tuple_einsum_test(num_inputs: int) → afe.backends.backend_checker.Predicate

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.reshape_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 = ['conv2d', 'conv2d_clip', 'conv2d_relu', 'conv2d_add', 'conv2d_add_clip', 'conv2d_add_relu',...

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

afe.backends.mla.mla_checkers.MLA_EXEMPT_FROM_BASELINE_CHECKS

afe.backends.mla.mla_checkers.MLA_5D_OPS

class afe.backends.mla.mla_checkers.MLAChecker

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

class afe.backends.mla.mla_checkers.RestrictedDimensionMLAChecker

Checker class for MLA backend, like MLAChecker except that it does not allow reshape on MLA. See BaseChecker for detailed documentation.