Searched refs:tensor_stat (Results 1 – 2 of 2) sorted by relevance
121 for x, (tensor_info_item, tensor_base, tensor_stat) in enumerate(zip(tensor_info,137 assert tensor_stat.data_size == stat_json['size_in_bytes']138 assert tensor_stat.dtype == stat_json['debugger_dtype']139 assert tensor_stat.shape == stat_json['shape']140 assert tensor_stat.is_bool == stat_json['is_bool']141 assert tensor_stat.max_value == stat_json['max_vaue']142 assert tensor_stat.min_value == stat_json['min_value']143 assert tensor_stat.avg_value == stat_json['avg_value']144 assert tensor_stat.count == stat_json['count']145 assert tensor_stat.neg_zero_count == stat_json['neg_zero_count'][all …]
934 void AddTensorStatInfo(const DebugServices::TensorStat &tensor_stat, in AddTensorStatInfo() argument942 tensor_base->set_data_type(tensor_stat.dtype); in AddTensorStatInfo()943 tensor_base->set_data_size((int64_t)tensor_stat.data_size); in AddTensorStatInfo()944 for (auto elem : tensor_stat.shape) { in AddTensorStatInfo()949 tensor_statistics->set_is_bool(tensor_stat.is_bool); in AddTensorStatInfo()950 tensor_statistics->set_max_value(static_cast<float>(tensor_stat.max_value)); in AddTensorStatInfo()951 tensor_statistics->set_min_value(static_cast<float>(tensor_stat.min_value)); in AddTensorStatInfo()952 tensor_statistics->set_avg_value(static_cast<float>(tensor_stat.avg_value)); in AddTensorStatInfo()953 tensor_statistics->set_count(tensor_stat.count); in AddTensorStatInfo()954 tensor_statistics->set_neg_zero_count(tensor_stat.neg_zero_count); in AddTensorStatInfo()[all …]