Searched refs:MeTensor (Results 1 – 9 of 9) sorted by relevance
41 using MeTensor = mindspore::tensor::Tensor; variable42 using MeTensorPtr = std::shared_ptr<MeTensor>;55 using TensorMap = std::unordered_map<std::string, std::shared_ptr<MeTensor>>;
182 if (value->isa<MeTensor>()) { in ConvertAnyUtil()184 return ConvertAnyUtil(value, AnyTraits<MeTensor>()); in ConvertAnyUtil()
346 MeTensor me_tensor(me_type, me_dims); in GenerateMeTensor()370 return make_shared<MeTensor>(me_tensor); in GenerateMeTensor()
1850 if (vec[i]->isa<MeTensor>()) { in TryConvertValueNodeToMultiConst()
39 void PrintMeTensor(MeTensor* tensor);
112 MeTensor tensor = MeTensor(TypeId::kNumberTypeFloat32, std::vector<int64_t>({1, 2, 3})); in TEST_F()
51 void PrintMeTensor(MeTensor* tensor) { in PrintMeTensor()
629 MeTensor me_tensor(type_id, dims); in TEST_F()634 auto me_tensor_ptr = std::make_shared<MeTensor>(me_tensor); in TEST_F()652 auto me_tensor_ptr = std::make_shared<MeTensor>(type_id, dims); in TEST_F()659 auto me_tensor_ptr_2 = std::make_shared<MeTensor>(type_id_2, dims2); in TEST_F()718 MeTensor& me_tensor = *me_tensor_ptr; in TEST_F()
39 using MeTensor = mindspore::tensor::Tensor; typedef59 if (py::isinstance<MeTensor>(obj)) { in ConvertPyObjToTensor()