Home
last modified time | relevance | path

Searched refs:MatrixDescriptor (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc74 struct MatrixDescriptor { struct
83 int64 batch_size, MatrixDescriptor lhs_matrix, MatrixDescriptor rhs_matrix, in DoGemmWithAlgorithm()
84 MatrixDescriptor output_matrix, AlphaType alpha, double beta, in DoGemmWithAlgorithm()
233 bool transpose) -> MatrixDescriptor { in RunGemm()
237 return MatrixDescriptor{ in RunGemm()
243 MatrixDescriptor lhs_matrix = make_descriptor( in RunGemm()
245 MatrixDescriptor rhs_matrix = make_descriptor( in RunGemm()
257 const MatrixDescriptor output_matrix{output_buffer, /*needs_transpose=*/false, in RunGemm()