Home
last modified time | relevance | path

Searched refs:input_data_layout (Results 1 – 3 of 3) sorted by relevance

/external/ComputeLibrary/src/core/CL/kernels/
DCLConvertFullyConnectedWeightsKernel.cpp63 …const DataLayout input_data_layout = (data_layout == DataLayout::NCHW) ? DataLayout::NHWC : DataLa… in configure() local
65 …const int width_idx = get_data_layout_dimension_index(input_data_layout, DataLayoutDimension::WI… in configure()
66 …const int height_idx = get_data_layout_dimension_index(input_data_layout, DataLayoutDimension::HE… in configure()
67 …const int channel_idx = get_data_layout_dimension_index(input_data_layout, DataLayoutDimension::CH… in configure()
/external/ComputeLibrary/src/core/NEON/kernels/
DNEConvertFullyConnectedWeightsKernel.cpp51 …const DataLayout input_data_layout = (data_layout == DataLayout::NCHW) ? DataLayout::NHWC : DataLa… in configure() local
53 …const int width_idx = get_data_layout_dimension_index(input_data_layout, DataLayoutDimension::WI… in configure()
54 …const int height_idx = get_data_layout_dimension_index(input_data_layout, DataLayoutDimension::HE… in configure()
55 …const int channel_idx = get_data_layout_dimension_index(input_data_layout, DataLayoutDimension::CH… in configure()
/external/ComputeLibrary/src/graph/
DGraphBuilder.cpp234 const DataLayout input_data_layout = input_tensor_desc.layout; in add_convolution_node() local
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()
242 w_desc.shape.set(get_dimension_idx(input_data_layout, DataLayoutDimension::BATCHES), depth); in add_convolution_node()
289 const DataLayout input_data_layout = input_tensor_desc.layout; in add_deconvolution_node() local
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()
297 w_desc.shape.set(get_dimension_idx(input_data_layout, DataLayoutDimension::BATCHES), depth); in add_deconvolution_node()
[all …]