Searched defs:TensorStat (Results 1 – 2 of 2) sorted by relevance
235 struct TensorStat { struct257 TensorStat() = default; argument259 uint64_t data_size = 0;260 int dtype = 0;261 std::vector<int64_t> shape;262 bool is_bool = false;265 double avg_value = 0.0;266 uint64_t count = 0;267 uint64_t neg_zero_count = 0;268 uint64_t pos_zero_count = 0;[all …]
40 …TensorStat(const string &type, const string &name, size_t task_id, size_t stream_id, uint64_t time… in TensorStat() function