MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / TRITONTF_ShapeNew

Function TRITONTF_ShapeNew

addons/triton/tensorflow_backend_tf.cc:638–650  ·  view source on GitHub ↗

TRITONTF_Shape

Source from the content-addressed store, hash-verified

636// TRITONTF_Shape
637//
638TRITONTF_Shape*
639TRITONTF_ShapeNew(size_t rank, int64_t* dims)
640{
641 TRITONTF_Shape* shape = new TRITONTF_Shape;
642 shape->rank_ = rank;
643 shape->dims_ = nullptr;
644 if (rank > 0) {
645 shape->dims_ = new int64_t[rank];
646 memcpy(shape->dims_, dims, rank * sizeof(int64_t));
647 }
648
649 return shape;
650}
651
652void
653TRITONTF_ShapeDelete(TRITONTF_Shape* shape)

Callers 2

TRITONTF_TensorNewFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected