Searched defs:MatrixDescriptor (Results 1 – 1 of 1) sorted by relevance
34 struct MatrixDescriptor { struct35 MatrixDescriptor(se::DeviceMemoryBase matrix_data, bool needs_transpose, in MatrixDescriptor() function44 se::DeviceMemoryBase data;45 bool transpose; // Whether this matrix needs to be transposed.46 int64 num_rows;47 int64 num_cols;48 int64 batch_size;