Searched refs:compute_type (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1064 cudnnDataType_t compute_type, in CudnnRnnDescriptor() argument
1080 compute_type_(compute_type), in CudnnRnnDescriptor()
1092 cudnnDataType_t data_type, cudnnDataType_t compute_type, in Create() argument
1142 /*dataType=*/data_type, /*mathPrec=*/compute_type, in Create()
1155 /*dataType=*/compute_type)); in Create()
1197 data_type, compute_type, algorithm_config, in Create()
1211 cudnnDataType_t compute_type() const { return compute_type_; } in compute_type() function in stream_executor::gpu::CudnnRnnDescriptor
|
| D | cuda_blas.cc | 443 blas::ComputationType compute_type) { in GetScaleType() argument
446 switch (compute_type) { in GetScaleType()
2459 cudaDataType_t compute_type = in DoBlasInternalImpl() local
2473 batch_count, compute_type, algo); in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3483 auto compute_type = RankedTensorType::get( in matchAndRewrite() local
3495 rewriter.create<mhlo::ConvertOp>(op.getLoc(), compute_type, abs); in matchAndRewrite()
3497 rewriter.create<mhlo::ConvertOp>(op.getLoc(), compute_type, delta); in matchAndRewrite()
|