Home
last modified time | relevance | path

Searched refs:DataLayoutDimension (Results 1 – 25 of 127) sorted by relevance

123456

/external/ComputeLibrary/src/core/
DHelpers.cpp32 …const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in calculate_valid_region_scale()
33 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in calculate_valid_region_scale()
104 const std::map<DataLayout, std::vector<DataLayoutDimension>> &get_layout_map() in get_layout_map()
106 constexpr DataLayoutDimension W = DataLayoutDimension::WIDTH; in get_layout_map()
107 constexpr DataLayoutDimension H = DataLayoutDimension::HEIGHT; in get_layout_map()
108 constexpr DataLayoutDimension C = DataLayoutDimension::CHANNEL; in get_layout_map()
109 constexpr DataLayoutDimension D = DataLayoutDimension::DEPTH; in get_layout_map()
110 constexpr DataLayoutDimension N = DataLayoutDimension::BATCHES; in get_layout_map()
112 static const std::map<DataLayout, std::vector<DataLayoutDimension>> layout_map = in get_layout_map()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h90 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in compute_vector_to_tensor_output_shape()
91 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in compute_vector_to_tensor_output_shape()
92 const size_t idx_c = get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL); in compute_vector_to_tensor_output_shape()
125 …e_t idx_width = get_data_layout_dimension_index(input.data_layout(), DataLayoutDimension::WIDTH); in compute_reorg_output_shape()
126 …_t idx_height = get_data_layout_dimension_index(input.data_layout(), DataLayoutDimension::HEIGHT); in compute_reorg_output_shape()
127 …t idx_channel = get_data_layout_dimension_index(input.data_layout(), DataLayoutDimension::CHANNEL); in compute_reorg_output_shape()
379 …const int width_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W…
380 …const int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H…
381 …const int channel_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::C…
427 …const int width_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in compute_depthwise_convolution_shape()
[all …]
/external/ComputeLibrary/src/graph/nodes/
DFusedDepthwiseConvolutionBatchNormalizationNode.cpp89 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDT… in compute_output_descriptor()
90 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIG… in compute_output_descriptor()
91 …const unsigned int input_channels = get_dimension_size(input_descriptor, DataLayoutDimension::CHAN… in compute_output_descriptor()
92 …const unsigned int kernel_width = get_dimension_size(weights_descriptor, DataLayoutDimension::WI… in compute_output_descriptor()
93 …const unsigned int kernel_height = get_dimension_size(weights_descriptor, DataLayoutDimension::HE… in compute_output_descriptor()
98 …output_descriptor.shape.set(get_dimension_idx(output_descriptor.layout, DataLayoutDimension::WIDTH… in compute_output_descriptor()
99 …output_descriptor.shape.set(get_dimension_idx(output_descriptor.layout, DataLayoutDimension::HEIGH… in compute_output_descriptor()
100 …output_descriptor.shape.set(get_dimension_idx(output_descriptor.layout, DataLayoutDimension::CHANN… in compute_output_descriptor()
DDepthwiseConvolutionLayerNode.cpp86 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDT… in compute_output_descriptor()
87 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIG… in compute_output_descriptor()
88 …const unsigned int input_channels = get_dimension_size(input_descriptor, DataLayoutDimension::CHAN… in compute_output_descriptor()
89 …const unsigned int kernel_width = get_dimension_size(weights_descriptor, DataLayoutDimension::WI… in compute_output_descriptor()
90 …const unsigned int kernel_height = get_dimension_size(weights_descriptor, DataLayoutDimension::HE… in compute_output_descriptor()
96 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::WIDTH), output_wid… in compute_output_descriptor()
97 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::HEIGHT), output_he… in compute_output_descriptor()
98 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::CHANNEL), input_ch… in compute_output_descriptor()
DDeconvolutionLayerNode.cpp54 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
55 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
56 …const unsigned int kernel_width = get_dimension_size(weights_descriptor, DataLayoutDimension::WID… in compute_output_descriptor()
57 …const unsigned int kernel_height = get_dimension_size(weights_descriptor, DataLayoutDimension::HEI… in compute_output_descriptor()
63 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::WIDTH), output_wid… in compute_output_descriptor()
64 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::HEIGHT), output_he… in compute_output_descriptor()
65 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::CHANNEL), weights_… in compute_output_descriptor()
DFusedConvolutionBatchNormalizationWithPostOpsNode.cpp87 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
88 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
89 …const unsigned int kernel_width = get_dimension_size(weights_descriptor, DataLayoutDimension::WID… in compute_output_descriptor()
90 …const unsigned int kernel_height = get_dimension_size(weights_descriptor, DataLayoutDimension::HEI… in compute_output_descriptor()
96 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::WIDTH), output_wid… in compute_output_descriptor()
97 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::HEIGHT), output_he… in compute_output_descriptor()
98 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::CHANNEL), weights_… in compute_output_descriptor()
DReorgLayerNode.cpp48 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
49 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
50 …const unsigned int input_channel = get_dimension_size(input_descriptor, DataLayoutDimension::CHANN… in compute_output_descriptor()
58 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::WIDTH), input_widt… in compute_output_descriptor()
59 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::HEIGHT), input_hei… in compute_output_descriptor()
60 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::CHANNEL), input_ch… in compute_output_descriptor()
DConvolutionLayerNode.cpp98 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
99 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
100 …const unsigned int kernel_width = get_dimension_size(weights_descriptor, DataLayoutDimension::WID… in compute_output_descriptor()
101 …const unsigned int kernel_height = get_dimension_size(weights_descriptor, DataLayoutDimension::HEI… in compute_output_descriptor()
107 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::WIDTH), output_wid… in compute_output_descriptor()
108 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::HEIGHT), output_he… in compute_output_descriptor()
109 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::CHANNEL), weights_… in compute_output_descriptor()
DFusedConvolutionBatchNormalizationNode.cpp98 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
99 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
100 …const unsigned int kernel_width = get_dimension_size(weights_descriptor, DataLayoutDimension::WID… in compute_output_descriptor()
101 …const unsigned int kernel_height = get_dimension_size(weights_descriptor, DataLayoutDimension::HEI… in compute_output_descriptor()
107 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::WIDTH), output_wid… in compute_output_descriptor()
108 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::HEIGHT), output_he… in compute_output_descriptor()
109 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::CHANNEL), weights_… in compute_output_descriptor()
DFusedConvolutionWithPostOpNode.cpp98 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
99 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
100 …const unsigned int kernel_width = get_dimension_size(weights_descriptor, DataLayoutDimension::WID… in compute_output_descriptor()
101 …const unsigned int kernel_height = get_dimension_size(weights_descriptor, DataLayoutDimension::HEI… in compute_output_descriptor()
107 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::WIDTH), output_wid… in compute_output_descriptor()
108 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::HEIGHT), output_he… in compute_output_descriptor()
109 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::CHANNEL), weights_… in compute_output_descriptor()
DROIAlignLayerNode.cpp72 …const size_t idx_n = get_data_layout_dimension_index(output_desc.layout, DataLayoutDimension::BATC… in configure_output()
73 …const size_t idx_c = get_data_layout_dimension_index(output_desc.layout, DataLayoutDimension::CHAN… in configure_output()
74 …const size_t idx_h = get_data_layout_dimension_index(output_desc.layout, DataLayoutDimension::HEIG… in configure_output()
75 …const size_t idx_w = get_data_layout_dimension_index(output_desc.layout, DataLayoutDimension::WIDT… in configure_output()
DPoolingLayerNode.cpp53 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH); in compute_output_descriptor()
54 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGHT… in compute_output_descriptor()
62 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::WIDTH), pooled_wid… in compute_output_descriptor()
63 …output_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::HEIGHT), pooled_he… in compute_output_descriptor()
/external/ComputeLibrary/src/core/CL/kernels/
DCLSpaceToDepthLayerKernel.cpp50 … int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_arguments()
51 …int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_arguments()
52 …nt idx_channel = get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL); in validate_arguments()
53 …nt idx_batch = get_data_layout_dimension_index(data_layout, DataLayoutDimension::BATCHES); in validate_arguments()
91 …idth = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDimension::WIDTH); in configure()
92 …nnel = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDimension::CHANNEL); in configure()
DCLDeconvolutionReshapeOutputKernel.cpp47 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_arguments()
48 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_arguments()
49 const size_t idx_b = get_data_layout_dimension_index(data_layout, DataLayoutDimension::BATCHES); in validate_arguments()
96 …ze_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_and_configure_window()
97 …e_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_and_configure_window()
138 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in configure()
139 …const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in configure()
140 …const size_t idx_b = get_data_layout_dimension_index(data_layout, DataLayoutDimension::B… in configure()
DCLDepthToSpaceLayerKernel.cpp47 …const int idx_channel = get_data_layout_dimension_index(data_layout, DataLayoutDimension::C… in validate_arguments()
53 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_arguments()
54 … const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_arguments()
91 …idth = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDimension::WIDTH); in configure()
92 …nnel = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDimension::CHANNEL); in configure()
DCLDeconvolutionLayerUpsampleKernel.cpp54 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate()
55 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate()
56 const size_t idx_c = get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL); in validate()
110 const size_t idx_w = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::WIDTH); in run()
111 const size_t idx_h = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::HEIGHT); in run()
DCLBatchToSpaceLayerKernel.cpp63 …const int idx_batch = get_data_layout_dimension_index(data_layout, DataLayoutDimension::B… in validate_arguments_static()
69 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_arguments_static()
70 … const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_arguments_static()
71 …const int idx_channel = get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL); in validate_arguments_static()
106 …_width = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDimension::WIDTH); in configure()
139 …_width = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDimension::WIDTH); in configure()
/external/ComputeLibrary/arm_compute/graph/
DLayerDescriptors.h41 : axis(DataLayoutDimension::CHANNEL), output_qinfo() in ConcatLayerDescriptor()
49 ConcatLayerDescriptor(DataLayoutDimension axis) in ConcatLayerDescriptor()
59 ConcatLayerDescriptor(DataLayoutDimension axis, QuantizationInfo output_qinfo) in ConcatLayerDescriptor()
64 const DataLayoutDimension axis; /**< Concatenation Axis */
/external/ComputeLibrary/src/graph/
DUtils.cpp142 size_t get_dimension_size(const TensorDescriptor &descriptor, const DataLayoutDimension data_layout… in get_dimension_size()
148 size_t get_dimension_idx(DataLayout data_layout, const DataLayoutDimension data_layout_dimension) in get_dimension_idx()
159 case DataLayoutDimension::CHANNEL: in get_dimension_idx()
162 case DataLayoutDimension::HEIGHT: in get_dimension_idx()
165 case DataLayoutDimension::WIDTH: in get_dimension_idx()
168 case DataLayoutDimension::BATCHES: in get_dimension_idx()
DGraphBuilder.cpp162 …hape = TensorShape(get_dimension_size(input_tensor_desc, DataLayoutDimension::CHANNEL)); in add_batch_normalization_node()
238 …w_desc.shape.set(get_dimension_idx(input_data_layout, DataLayoutDimension::WIDTH), kernel_spatial_… in add_convolution_node()
239 …w_desc.shape.set(get_dimension_idx(input_data_layout, DataLayoutDimension::HEIGHT), kernel_spatial… in add_convolution_node()
240 w_desc.shape.set(get_dimension_idx(input_data_layout, DataLayoutDimension::CHANNEL), in add_convolution_node()
241 … get_dimension_size(input_tensor_desc, DataLayoutDimension::CHANNEL) / num_groups); in add_convolution_node()
242 w_desc.shape.set(get_dimension_idx(input_data_layout, DataLayoutDimension::BATCHES), depth); in add_convolution_node()
293 …w_desc.shape.set(get_dimension_idx(input_data_layout, DataLayoutDimension::WIDTH), kernel_spatial_… in add_deconvolution_node()
294 …w_desc.shape.set(get_dimension_idx(input_data_layout, DataLayoutDimension::HEIGHT), kernel_spatial… in add_deconvolution_node()
295 w_desc.shape.set(get_dimension_idx(input_data_layout, DataLayoutDimension::CHANNEL), in add_deconvolution_node()
296 get_dimension_size(input_tensor_desc, DataLayoutDimension::CHANNEL)); in add_deconvolution_node()
[all …]
/external/ComputeLibrary/src/core/NEON/kernels/
DNEReorgLayerKernel.cpp49 …e_t idx_width = get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::WIDTH); in validate_arguments()
50 …_t idx_height = get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::HEIGHT); in validate_arguments()
109 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in run()
110 …const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in run()
111 …const size_t idx_c = get_data_layout_dimension_index(data_layout, DataLayoutDimension::C… in run()
DNESpaceToDepthLayerKernel.cpp56 … int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_arguments()
57 …int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_arguments()
58 …nt idx_channel = get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL); in validate_arguments()
59 …nt idx_batch = get_data_layout_dimension_index(data_layout, DataLayoutDimension::BATCHES); in validate_arguments()
108 …const int channel_idx = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::CHANNE… in run()
/external/ComputeLibrary/src/cpu/kernels/
DCpuPool3dKernel.cpp80 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_arguments()
81 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_arguments()
82 … const int idx_depth = get_data_layout_dimension_index(data_layout, DataLayoutDimension::DEPTH); in validate_arguments()
133 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in configure()
134 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in configure()
135 … const int idx_depth = get_data_layout_dimension_index(data_layout, DataLayoutDimension::DEPTH); in configure()
/external/ComputeLibrary/src/runtime/heuristics/dwc_native/
DClDWCNativeDefaultConfigValhall.cpp78 …idx_c = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimension::CHANNEL); in configure_G78_f32()
79 …t idx_w = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimension::WIDTH); in configure_G78_f32()
141 …idx_c = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimension::CHANNEL); in configure_G78_f16()
142 …t idx_w = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimension::WIDTH); in configure_G78_f16()
244 …idx_c = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimension::CHANNEL); in configure_G77_f16()
245 …t idx_w = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimension::WIDTH); in configure_G77_f16()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEDeconvolutionLayer.cpp89 …t width_idx = get_data_layout_dimension_index(weights->data_layout(), DataLayoutDimension::WIDTH); in validate()
90 … height_idx = get_data_layout_dimension_index(weights->data_layout(), DataLayoutDimension::HEIGHT); in validate()
134 …ize_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate()
135 …ze_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate()
147 …atches_idx = get_data_layout_dimension_index(weights->data_layout(), DataLayoutDimension::BATCHES); in validate()
148 …hannel_idx = get_data_layout_dimension_index(weights->data_layout(), DataLayoutDimension::CHANNEL); in validate()
165 …nsigned int width_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in configure()
166 …signed int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in configure()

123456