Method IsScalar

tensorflow/stream_executor/device_memory.h:188–188

Returns whether this is a single-element allocation.

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

// Host-side representation of packed-and-aligned vector datatypes on the device