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

Function JoinedShape

dali/kernels/common/join/tensor_join_shape.h:56–58  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

54}
55
56inline TensorShape<> JoinedShape(span<const TensorShape<>> in_shapes, int axis, bool new_axis) {
57 return new_axis ? ConcatenatedShape(in_shapes, axis) : StackedShape(in_shapes, axis);
58}
59
60
61template <typename GetInListShapeFunc>

Callers 3

SetupTypedMethod · 0.85
SetupMethod · 0.85
TESTFunction · 0.85

Calls 9

ConcatenatedShapeFunction · 0.85
StackedShapeFunction · 0.85
joined_ndimFunction · 0.85
CheckJoinedShapesFunction · 0.85
resizeMethod · 0.45
sample_dimMethod · 0.45
num_samplesMethod · 0.45
num_elementsMethod · 0.45
tensor_shape_spanMethod · 0.45

Tested by 1

TESTFunction · 0.68