Searched refs:input_data_layout (Results 1 – 3 of 3) sorted by relevance
/external/ComputeLibrary/src/core/CL/kernels/ |
D | CLConvertFullyConnectedWeightsKernel.cpp | 63 …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/ |
D | NEConvertFullyConnectedWeightsKernel.cpp | 51 …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/ |
D | GraphBuilder.cpp | 234 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 …]
|