Add a default constructor to Array4D
PiperOrigin-RevId: 293904686 Change-Id: I6520442ad0e808a1c6b8a5ea13e2849a06c00d1e
This commit is contained in:
parent
8ff354d254
commit
74c743506d
@ -54,6 +54,8 @@ namespace xla {
|
||||
template <typename T>
|
||||
class Array4D : public Array<T> {
|
||||
public:
|
||||
Array4D() : Array<T>(std::vector<int64>{0, 0, 0, 0}) {}
|
||||
|
||||
// Creates a 4D array, uninitialized values.
|
||||
Array4D(int64 planes, int64 depth, int64 height, int64 width)
|
||||
: Array<T>(std::vector<int64>{planes, depth, height, width}) {}
|
||||
|
Loading…
Reference in New Issue
Block a user