Searched refs:MatrixDescriptor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 34 struct MatrixDescriptor { struct
35 MatrixDescriptor(se::DeviceMemoryBase matrix_data, bool needs_transpose, in MatrixDescriptor() function
54 bool DoGemm(MatrixDescriptor lhs_matrix, MatrixDescriptor rhs_matrix, in DoGemm()
55 MatrixDescriptor output_matrix, double alpha, double beta, in DoGemm()
114 bool DoGemmWithAlgorithm(MatrixDescriptor lhs_matrix, in DoGemmWithAlgorithm()
115 MatrixDescriptor rhs_matrix, in DoGemmWithAlgorithm()
116 MatrixDescriptor output_matrix, double alpha, in DoGemmWithAlgorithm()
158 MatrixDescriptor lhs_matrix, MatrixDescriptor rhs_matrix, in DoGemmAutotune()
159 MatrixDescriptor output_matrix, double alpha, double beta, in DoGemmAutotune()
376 bool transpose) -> MatrixDescriptor { in ExecuteOnStream()
[all …]
|