Searched refs:weights_height_idx (Results 1 – 3 of 3) sorted by relevance
135 …const int weights_height_idx = get_data_layout_dimension_index(weight_layout, DataLayoutDimension… in try_in_place_depthwiseconv() local136 …_1x1 = weight_shape[weights_width_idx] == 1U && weight_shape[weights_height_idx] == 1U; in try_in_place_depthwiseconv()
76 …const int weights_height_idx = get_data_layout_dimension_index(weights->data_layout(), DataLayoutD… in validate_arguments() local77 …hts->tensor_shape()[weights_width_idx] != 1U || weights->tensor_shape()[weights_height_idx] != 1U); in validate_arguments()
433 …const int weights_height_idx = get_data_layout_dimension_index(weights_data_layout, DataLa… in compute_depthwise_convolution_shape() local438 … weights_shape[weights_width_idx], weights_shape[weights_height_idx], in compute_depthwise_convolution_shape()