experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add utility fct FillWithZY

Browse Source
This commit is contained in:
Frederic Bastien 2019-10-08 12:28:47 -07:00
parent 90629a96fd
commit bace8b06d0
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tensorflow/compiler/xla/array4d.h
View File

@ -120,6 +120,21 @@ class Array4D : public Array<T> {
}
}
// Fills all of the {p,x} with the array provided, which specifies {z,y}.
void FillWithZY(const Array2D<T>& value) {
CHECK_EQ(value.height(), depth());
CHECK_EQ(value.width(), height());
for (int64 plane = 0; plane < planes(); ++plane) {
for (int64 depth = 0; depth < this->depth(); ++depth) {
for (int64 height = 0; height < this->height(); ++height) {
for (int64 width = 0; width < this->width(); ++width) {
(*this)(plane, depth, height, width) = value(depth, height);
}
}
}
}
}
// Fills all of the {x,y} with the array provided, which specifies {p,z}.
void FillWithPZ(const Array2D<T>& value) {
CHECK_EQ(value.height(), planes());