Searched refs:lhs_matrix (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 83 int64 batch_size, MatrixDescriptor lhs_matrix, MatrixDescriptor rhs_matrix, in DoGemmWithAlgorithm() argument
117 se::DeviceMemory<Element> lhs_data(lhs_matrix.data); in DoGemmWithAlgorithm()
121 auto lhs_transpose = lhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm()
125 auto k = lhs_matrix.transpose ? lhs_matrix.num_rows : lhs_matrix.num_cols; in DoGemmWithAlgorithm()
136 /*leading dim of LHS=*/lhs_matrix.num_rows, rhs_data, in DoGemmWithAlgorithm()
145 int64 lhs_stride = lhs_matrix.num_rows * lhs_matrix.num_cols; in DoGemmWithAlgorithm()
153 /*leading dim of LHS=*/lhs_matrix.num_rows, lhs_stride, rhs_data, in DoGemmWithAlgorithm()
165 lhs_data, /*leading dim of LHS=*/lhs_matrix.num_rows, rhs_data, in DoGemmWithAlgorithm()
243 MatrixDescriptor lhs_matrix = make_descriptor( in RunGemm() local
253 std::swap(lhs_matrix, rhs_matrix); in RunGemm()
[all …]
|