Lines Matching refs:data_layout
64 const DataLayout data_layout = input->data_layout(); in validate() local
66 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate()
67 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate()
68 const size_t idx_c = get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL); in validate()
106 …is_resizable(true).reset_padding().set_tensor_shape(scale_out_shape).set_data_layout(data_layout)); in validate()
134 const DataLayout data_layout = input->info()->data_layout(); in configure() local
136 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in configure()
137 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in configure()
141 _weights_flipped.allocator()->init(weights->info()->clone()->set_data_layout(data_layout)); in configure()
149 …tput->info(), input->info()->clone()->set_tensor_shape(output_shape).set_data_layout(data_layout)); in configure()
178 scale_out_info.set_data_layout(data_layout); in configure()
194 if(weights->info()->data_layout() == DataLayout::NHWC) in configure()