Searched refs:rhs_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
118 se::DeviceMemory<Element> rhs_data(rhs_matrix.data); in DoGemmWithAlgorithm()
123 auto rhs_transpose = rhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm()
137 /*leading dim of RHS=*/rhs_matrix.num_rows, in DoGemmWithAlgorithm()
146 int64 rhs_stride = rhs_matrix.num_rows * rhs_matrix.num_cols; in DoGemmWithAlgorithm()
154 /*leading dim of RHS=*/rhs_matrix.num_rows, rhs_stride, in DoGemmWithAlgorithm()
166 /*leading dim of RHS=*/rhs_matrix.num_rows, /*beta=*/beta, in DoGemmWithAlgorithm()
245 MatrixDescriptor rhs_matrix = make_descriptor( in RunGemm() local
253 std::swap(lhs_matrix, rhs_matrix); in RunGemm()
278 batch_size, lhs_matrix, rhs_matrix, output_matrix, alpha.real(), in RunGemm()
[all …]
|