Searched defs:TensorTraits (Results 1 – 1 of 1) sorted by relevance
244 struct TensorTraits<float> { struct245 static int size(const TensorProto& p) { return p.float_val_size(); } in size()246 static float get(const TensorProto& p, int i) { return p.float_val(i); } in get()247 static string accessor_name() { return "float_val"; } in accessor_name()248 static string type_name() { return "float"; } in type_name()249 static void CopyFromContent(const TensorProto& p, std::vector<float>* data) { in CopyFromContent()256 struct TensorTraits<uint8_t> { struct257 static int size(const TensorProto& p) { return p.int_val_size(); } in size()258 static uint8_t get(const TensorProto& p, int i) { return p.int_val(i); } in get()259 static string accessor_name() { return "int_val"; } in accessor_name()[all …]