| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/bert/bert_performance/src/ |
| D | bert_model.py | 73 compute_type=mstype.float32): argument
229 compute_type=mstype.float32): argument
399 compute_type=mstype.float32): argument
570 compute_type=mstype.float32): argument
632 compute_type=mstype.float32): argument
694 compute_type=mstype.float32, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/cuda_impl/ |
| D | cublas_utils.cc | 42 cudaDataType compute_type = data_types[3]; in CublasMM1Batch() local
64 cudaDataType compute_type = data_types[3]; in CublasMMBatched() local
86 cublasComputeType_t compute_type = CUBLAS_COMPUTE_32F_FAST_TF32; in CublasGemmWrapper() local
109 cublasComputeType_t compute_type = CUBLAS_COMPUTE_32F_FAST_TF32; in CublasGemmStridedBatchedWrapper() local
146 cublasComputeType_t compute_type = CUBLAS_COMPUTE_32F_FAST_TF32; in CublasLtGemmWrapper() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/bert/src/ |
| D | bert_model.py | 80 compute_type=mstype.float32, argument
245 compute_type=mstype.float32, argument
427 compute_type=mstype.float32): argument
612 compute_type=mstype.float32, argument
684 compute_type=mstype.float32, argument
754 compute_type=mstype.float32, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/matmul/ |
| D | matmul_wrapper.cc | 36 cublasComputeType_t compute_type = data_compute_type_map_[data_type]; in GetComputeType() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_autoparallel_rec_shared_param_strmodif.py | 55 compute_type=mstype.float16): argument
|
| D | test_auto_parallel_frontend_scheduling_with_pangu_alpha.py | 530 compute_type=mstype.float16): argument
|
| D | test_auto_parallel_pangu_alpha.py | 531 compute_type=mstype.float16): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | decoder_tensorrt.h | 56 DecoderPlugin(const std::string name, int compute_type, in DecoderPlugin()
|
| D | encoder_tensorrt.h | 60 EncoderPlugin(const std::string name, int compute_type, in EncoderPlugin()
|
| D | decoder_tensorrt.cc | 192 auto compute_type = runtime_->GetRuntimePrecisionMode(); in AddInnerOp() local
|
| D | encoder_tensorrt.cc | 276 auto compute_type = runtime_->GetRuntimePrecisionMode(); in AddInnerOp() local
|
| /third_party/mindspore/mindspore-src/source/tests/mindspore_test_framework/apps/ |
| D | bert_attention_submodules.py | 100 compute_type=mstype.float32): argument
292 compute_type=mstype.float32): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | dense_grad_gpu_kernel.cc | 63 cublasComputeType_t compute_type = CUBLAS_COMPUTE_16F; in GetComputeType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/ |
| D | op_node.h | 57 ComputeType compute_type() const { return compute_type_; } in compute_type() function
|