Method ElementCount

tensorflow/stream_executor/device_memory.h:138–138

Returns the number of elements of type ElemT that constitute this allocation.

  // Returns the number of elements of type ElemT that constitute this
  // allocation.
  uint64 ElementCount() const { return size() / sizeof(ElemT); }

  // Returns whether this is a single-element allocation.
  bool IsScalar() const { return ElementCount() == 1; }