Compute Library
18.05
|
Data Structures | |
class | BackendRegistrar |
Helper class to statically register a backend. More... | |
Functions | |
arm_compute::ITensorInfo * | get_backing_tensor_info (arm_compute::graph::Tensor *tensor) |
Returns backing tensor info of a given tensor. More... | |
template<typename ConvolutionLayer , typename DirectConvolutionLayer , typename GEMMConvolutionLayer , typename WinogradConvolutionLayer > | |
Status | validate_convolution_layer (ConvolutionLayerNode &node) |
Validates a Convolution layer node. More... | |
template<typename DepthwiseConvolutionLayer , typename DepthwiseConvolutionLayer3x3 > | |
Status | validate_depthwise_convolution_layer (DepthwiseConvolutionLayerNode &node) |
Validates a Depthwise Convolution layer node. More... | |
|
inline |
Returns backing tensor info of a given tensor.
[in] | tensor | Tensor to extract the backing tensor from |
Definition at line 50 of file ValidateHelpers.h.
References Tensor::handle(), ITensor::info(), and ITensorHandle::tensor().
Referenced by validate_convolution_layer(), and validate_depthwise_convolution_layer().
Status arm_compute::graph::backends::detail::validate_convolution_layer | ( | ConvolutionLayerNode & | node | ) |
Validates a Convolution layer node.
ConvolutionLayer | Default Convolution layer function type |
DirectConvolutionLayer | Direct Convolution layer function type |
GEMMConvolutionLayer | GEMM Convolution layer function type |
WinogradConvolutionLayer | Winograd Convolution layer function type |
[in] | node | Node to validate |
Definition at line 67 of file ValidateHelpers.h.
References ARM_COMPUTE_LOG_GRAPH_INFO, ARM_COMPUTE_LOG_GRAPH_VERBOSE, ARM_COMPUTE_RETURN_ERROR_ON, arm_compute::test::validation::conv_info, ConvolutionLayerNode::convolution_info(), ConvolutionLayerNode::convolution_method(), arm_compute::graph::DEFAULT, arm_compute::graph::DIRECT, arm_compute::graph::GEMM, get_backing_tensor_info(), INode::id(), INode::input(), arm_compute::is_data_type_quantized_asymmetric(), INode::name(), INode::num_inputs(), INode::num_outputs(), INode::output(), arm_compute::S32, ConvolutionLayerNode::set_convolution_method(), ITensorInfo::set_data_type(), arm_compute::test::validation::validate(), and arm_compute::graph::WINOGRAD.
Status arm_compute::graph::backends::detail::validate_depthwise_convolution_layer | ( | DepthwiseConvolutionLayerNode & | node | ) |
Validates a Depthwise Convolution layer node.
DepthwiseConvolutionLayer | Default Depthwise Convolution layer type |
DepthwiseConvolutionLayer3x3 | Optimized 3x3 Depthwise Convolution layer type |
[in] | node | Node to validate |
Definition at line 132 of file ValidateHelpers.h.
References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_LOG_GRAPH_INFO, ARM_COMPUTE_LOG_GRAPH_VERBOSE, ARM_COMPUTE_RETURN_ERROR_ON, arm_compute::graph::DEFAULT, DepthwiseConvolutionLayerNode::depthwise_convolution_method(), get_backing_tensor_info(), arm_compute::get_data_layout_dimension_index(), INode::id(), INode::input(), INode::name(), INode::num_inputs(), INode::num_outputs(), arm_compute::graph::OPTIMIZED_3x3, DepthwiseConvolutionLayerNode::set_depthwise_convolution_method(), and arm_compute::WIDTH.