Searched refs:matmul_repeats (Results 1 – 1 of 1) sorted by relevance
388 const int matmul_repeats = input_overall_size / weights_shape.dims(1); in ProcessFullyConnectedOperator() local389 CHECK_EQ(matmul_repeats * weights_shape.dims(1), input_overall_size); in ProcessFullyConnectedOperator()392 output_array.copy_shape(Shape({matmul_repeats, weights_output_depth})); in ProcessFullyConnectedOperator()