Searched defs:TensorTraits (Results 1 – 1 of 1) sorted by relevance
254 struct TensorTraits<float> { struct255 static int size(const TensorProto& p) { return p.float_val_size(); } in size()256 static float get(const TensorProto& p, int i) { return p.float_val(i); } in get()257 static std::string accessor_name() { return "float_val"; } in accessor_name()258 static std::string type_name() { return "float"; } in type_name()259 static void CopyFromContent(const TensorProto& p, std::vector<float>* data) { in CopyFromContent()266 struct TensorTraits<uint8_t> { struct267 static int size(const TensorProto& p) { return p.int_val_size(); } in size()268 static uint8_t get(const TensorProto& p, int i) { return p.int_val(i); } in get()269 static std::string accessor_name() { return "int_val"; } in accessor_name()[all …]