Searched defs:data_types (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/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
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/graph/acl/ |
| D | model_process.cc | 57 … std::vector<std::vector<int64_t>> *shapes, std::vector<enum DataType> *data_types, in ConstructTensorDesc()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/ascend310/src/ |
| D | model_process.cc | 61 … std::vector<std::vector<int64_t>> *shapes, std::vector<enum DataType> *data_types, in ConstructTensorDesc()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 132 auto data_types = AnfAlgo::GetOutputInferDataType(one_hot_node, 0); in CreateSoftmaxCrossEntropyWithLogits() local
|