Searched refs:ab_type (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 3273 a_desc_, CreateCublasLtLayoutDesc(p.ab_type, rows_a, cols_a, p.lda, in DoBlasInternalImpl()
3276 b_desc_, CreateCublasLtLayoutDesc(p.ab_type, rows_b, cols_b, p.ldb, in DoBlasInternalImpl()
3289 CreateCublasLtLayoutDesc(p.ab_type, rows_a, cols_a, p.lda, p.stride_a, in DoBlasInternalImpl()
3293 CreateCublasLtLayoutDesc(p.ab_type, rows_b, cols_b, p.ldb, p.stride_b, in DoBlasInternalImpl()
3327 blas::DataType ab_type() const override { return params_.ab_type; } in DoBlasInternalImpl() function in stream_executor::gpu::__anone9de0a230311::CUDABlasLtMatmulPlan
3419 cuda_plan.params().ab_type))); in DoBlasInternalImpl()
3425 cuda_plan.params().ab_type))); in DoBlasInternalImpl()
3624 int ab_type_size = GetDataTypeSizeBytes(cuda_plan.params().ab_type); in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 198 virtual DataType ab_type() const = 0;
215 DataType ab_type; member
1500 constexpr blas::DataType ab_type = blas::ToDataType<ABType>::value; variable
1501 if (ab_type != plan->ab_type()) {
1504 << plan->ab_type() << ", got " << ab_type;
|