Lines Matching refs:data_layout
159 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_lay… in validate_arguments() local
160 …const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimen… in validate_arguments()
161 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimen… in validate_arguments()
184 && (src->data_layout() == DataLayout::NHWC), in validate_arguments()
199 …get_implementation(PoolDataTypeISASelectorData{ src->data_type(), src->data_layout(), pool_stride_… in validate_arguments()
218 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_layout() : pool_… in validate_and_configure_window() local
222 …const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimen… in validate_and_configure_window()
223 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimen… in validate_and_configure_window()
287 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_layout() : pool_… in configure() local
288 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in configure()
289 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in configure()
299 …get_implementation(PoolDataTypeISASelectorData{ src->data_type(), src->data_layout(), (int)pad_str… in configure()
304 _data_layout = src->data_layout(); in configure()
335 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_layout() : pool_… in validate() local
336 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate()
337 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate()