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

Method tensor_shape_span

include/dali/core/tensor_shape.h:600–604  ·  view source on GitHub ↗

* @brief Return a span containing the shape of `sample` */

Source from the content-addressed store, hash-verified

598 * @brief Return a span containing the shape of `sample`
599 */
600 span<int64_t, sample_ndim == DynamicDimensions ?
601 dynamic_extent : span_extent_t(sample_ndim)>
602 tensor_shape_span(int64_t sample) {
603 return {&shapes[sample * sample_dim()], span_extent_t(sample_dim())};
604 }
605
606 span<const int64_t, sample_ndim == DynamicDimensions ?
607 dynamic_extent : span_extent_t(sample_ndim)>

Callers 9

calculate_pointersFunction · 0.45
is_uniformFunction · 0.45
is_degenerate_dimFunction · 0.45
collapse_dimFunction · 0.45
collapse_dimsFunction · 0.45
permute_samplesFunction · 0.45
permute_dimsFunction · 0.45
sample_rangeFunction · 0.45
unfold_outer_dimFunction · 0.45

Calls 1

sample_dimFunction · 0.85

Tested by

no test coverage detected