MCPcopy Create free account
hub / github.com/NVIDIA/DALI / ConvertShape

Function ConvertShape

dali/python/backend_impl.cc:204–207  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

202
203template<typename SrcBackend>
204const TensorListShape<> ConvertShape(const TensorShape<> &shape,
205 TensorList<SrcBackend> *shape_type_placeholder) {
206 return uniform_list_shape(shape[0], shape.last(shape.size()-1));
207}
208
209template<typename SrcBackend>
210const TensorShape<> &ConvertShape(const TensorShape<> &shape,

Callers 3

FillTensorFromDlPackFunction · 0.85
FillTensorFromCudaArrayFunction · 0.85
RunBackendMethod · 0.85

Calls 3

uniform_list_shapeFunction · 0.85
lastMethod · 0.45
sizeMethod · 0.45

Tested by

no test coverage detected