| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/cuda_impl/ |
| D | cublas_utils.cc | 30 …const cublasOperation_t *operations, const cudaDataType *data_types, cublasHandle_t cublas_handle)… in CublasMM1Batch()
50 const cublasOperation_t *operations, const cudaDataType *data_types, in CublasMMBatched()
73 … const cublasOperation_t *operations, const cudaDataType *data_types, void *alpha, void *beta, in CublasGemmWrapper()
96 … const cudaDataType *data_types, void *alpha, void *beta, int batch, in CublasGemmStridedBatchedWrapper()
123 … const cublasOperation_t *operations, const cudaDataType *data_types, void *alpha, void *beta, in CublasLtGemmWrapper()
|
| D | cudnn_utils.cc | 22 …std::unordered_map<nvinfer1::DataType, cudnnDataType_t> data_types = {{nvinfer1::DataType::kFLOAT,… in ConvertCudnnDataType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | gather_tests.cc | 105 …TypeId data_types[] = {kNumberTypeInt32, kNumberTypeInt64, kNumberTypeFloat32, kNumberTypeFloat16}; in TEST_F() local
136 …TypeId data_types[] = {kNumberTypeInt32, kNumberTypeInt64, kNumberTypeFloat32, kNumberTypeFloat16}; in TEST_F() local
167 …TypeId data_types[] = {kNumberTypeInt32, kNumberTypeInt64, kNumberTypeFloat32, kNumberTypeFloat16}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/om/ |
| D | om_model_parser.cc | 316 std::vector<TypeId> data_types; in ParseDataTypes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | embedding_cache_prefetch_actor.cc | 1187 std::vector<TypeId> data_types = {kNumberTypeInt32, kNumberTypeFloat32, kNumberTypeInt32}; in SendToRemote() local
1496 bool Sender::Send(const std::vector<ShapeVector> &shapes, const std::vector<TypeId> data_types, in Send()
1553 const std::vector<TypeId> data_types, in BuildRpcMessage()
1647 …t Sender::CalDataSize(const std::vector<ShapeVector> &shapes, const std::vector<TypeId> data_types, in CalDataSize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | all_to_all_v_for_ge.cc | 228 std::vector<TypeId> data_types(origin_output_shapes.size(), base_dtype); in CreateSplitNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/graph/acl/ |
| D | model_process.cc | 65 … std::vector<std::vector<int64_t>> *shapes, std::vector<enum DataType> *data_types, in ConstructTensorDesc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/ascend/model/ |
| D | model_process.cc | 263 std::vector<TypeId> data_types; in GetOutputDataType() local
288 std::vector<TypeId> data_types; in GetInputDataType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/ |
| D | ir_pass.cc | 204 auto data_types = common::AnfAlgo::GetOutputInferDataType(input_node, kIndex0); in CreateReshape() local
273 auto data_types = common::AnfAlgo::GetOutputInferDataType(one_hot_node, kIndex0); in CreateSoftmaxCrossEntropyWithLogits() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/ascend/src/ |
| D | model_process.cc | 69 … std::vector<std::vector<int64_t>> *shapes, std::vector<enum DataType> *data_types, in ConstructTensorDesc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 96 auto data_types = common::AnfAlgo::GetOutputInferDataType(input_node, 0UL); in CreateReshape() local
192 auto data_types = common::AnfAlgo::GetOutputInferDataType(one_hot_node, 0UL); in CreateSoftmaxCrossEntropyWithLogits() local
|