MCPcopy Create free account
hub / github.com/Meshcapade/difflocks / batches

Function batches

extensions/include/cuMat/src/UnaryOps.h:765–771  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

763 : 1;
764 }
765 __host__ __device__ CUMAT_STRONG_INLINE Index batches() const
766 {
767 return _Batch == Axis::Row ? child_.rows()
768 : _Batch == Axis::Column ? child_.cols()
769 : _Batch == Axis::Batch ? child_.batches()
770 : 1;
771 }
772
773 __device__ CUMAT_STRONG_INLINE Scalar coeff(Index row, Index col, Index batch, Index index) const
774 {

Calls 3

rowsMethod · 0.45
colsMethod · 0.45
batchesMethod · 0.45

Tested by

no test coverage detected