Searched defs:TensorBuilder (Results 1 – 2 of 2) sorted by relevance
338 struct TensorBuilder { struct339 typedef Tensor Table;340 flatbuffers::FlatBufferBuilder &fbb_;341 flatbuffers::uoffset_t start_;342 void add_nodeType(int32_t nodeType) { in add_nodeType()345 void add_dataType(int32_t dataType) { in add_dataType()348 void add_dims(flatbuffers::Offset<flatbuffers::Vector<int32_t>> dims) { in add_dims()351 void add_format(mindspore::schema::Format format) { in add_format()354 void add_refCount(int32_t refCount) { in add_refCount()378 explicit TensorBuilder(flatbuffers::FlatBufferBuilder &_fbb) in TensorBuilder() argument[all …]
476 struct TensorBuilder { struct477 typedef Tensor Table;478 flatbuffers::FlatBufferBuilder &fbb_;479 flatbuffers::uoffset_t start_;480 void add_nodeType(int32_t nodeType) { in add_nodeType()483 void add_dataType(int32_t dataType) { in add_dataType()486 void add_dims(flatbuffers::Offset<flatbuffers::Vector<int32_t>> dims) { in add_dims()489 void add_format(mindspore::schema::Format format) { in add_format()492 void add_refCount(int32_t refCount) { in add_refCount()516 explicit TensorBuilder(flatbuffers::FlatBufferBuilder &_fbb) in TensorBuilder() argument[all …]