Searched refs:output_conv2d_shape (Results 1 – 1 of 1) sorted by relevance
769 …TensorShape output_conv2d_shape = compute_deep_convolution_shape(input_shape, input_data_layout, w… in compute_indirect_buffer_shape() local772 const unsigned int output_h = DIV_CEIL(output_conv2d_shape[1] * output_conv2d_shape[2], m0); in compute_indirect_buffer_shape()773 const unsigned int output_b = output_conv2d_shape[3]; in compute_indirect_buffer_shape()