Searched defs:compute_type (Results 1 – 3 of 3) sorted by relevance
1064 cudnnDataType_t compute_type, in CudnnRnnDescriptor()1092 cudnnDataType_t data_type, cudnnDataType_t compute_type, in Create()1211 cudnnDataType_t compute_type() const { return compute_type_; } in compute_type() function in stream_executor::gpu::CudnnRnnDescriptor
443 blas::ComputationType compute_type) { in GetScaleType()2459 cudaDataType_t compute_type = in DoBlasInternalImpl() local
3483 auto compute_type = RankedTensorType::get( in matchAndRewrite() local