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

Function ConvertShape

addons/triton/tensorflow_backend_tf.cc:148–155  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

146}
147
148void
149ConvertShape(TRITONTF_Shape* shape, tensorflow::TensorShape* tfshape)
150{
151 for (size_t itr = 0; itr < shape->rank_; itr++) {
152 const int64_t dim = shape->dims_[itr];
153 tfshape->AddDim(dim);
154 }
155}
156
157TRITONTF_Shape*
158ConvertShape(const tensorflow::TensorShape& tfshape)

Callers 2

TensorImplMethod · 0.85
TRITONTF_TensorNewFunction · 0.85

Calls 3

AddDimMethod · 0.45
dimsMethod · 0.45
dim_sizeMethod · 0.45

Tested by

no test coverage detected