Searched refs:output_num_rows (Results 1 – 2 of 2) sorted by relevance
351 int64 output_num_rows = output_shape_.dimensions(row_dim); in ExecuteOnStream() local473 MatrixDescriptor(output_data, false, output_num_rows, in ExecuteOnStream()479 output_num_rows, batch_size), in ExecuteOnStream()
93 output_num_rows): argument117 self._output_num_rows = output_num_rows