Home
last modified time | relevance | path

Searched defs:CheckTensor (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dtensor_util.h94 struct CheckTensor { struct
102 std::string tensor_name;
103 std::vector<size_t> shape;
104 std::vector<float> data;
105 std::vector<std::string> strings_data;
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/quick_start_ios/mindspore-lite/
DBenchmark.hpp63 struct CheckTensor { struct
64 CheckTensor(const std::vector<size_t> &shape, const std::vector<float> &data, in CheckTensor() function
70 std::vector<size_t> shape;
71 std::vector<float> data;
72 std::vector<std::string> strings_data;
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/benchmark/
Dbenchmark_base.h104 struct MS_API CheckTensor { struct
111 std::vector<size_t> shape;
112 std::vector<float> data;
113 std::vector<std::string> strings_data;
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/probability/distribution/_utils/
Dutils.py306 class CheckTensor(PrimitiveWithInfer): class