Home
last modified time | relevance | path

Searched defs:compute_type (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/tests/st/networks/models/bert/src/
Dbert_model.py80 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/tests/mindspore_test_framework/apps/
Dbert_attention_submodules.py100 compute_type=mstype.float32): argument
292 compute_type=mstype.float32): argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dmatmul_gpu_kernel.h57 cudaDataType_t compute_type = (dtype_a_ == CUDA_R_64F) ? CUDA_R_64F : CUDA_R_32F; in Launch() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/
Dop_node.h61 ComputeType compute_type() const { return compute_type_; } in compute_type() function
/third_party/mindspore/tests/st/fl/albert/src/
Dmodel.py58 def __init__(self,
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Dimage_utils.cc868 int compute_type = CV_32F; in NormalizePad() local