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

[XLA] Fix formatting of code snippets in the operation_semantics

Browse Source 
PiperOrigin-RevId: 314401465
Change-Id: I43e11bc63a3b014f9ff621413ddf385d74f334a5
This commit is contained in:
David Majnemer 2020-06-02 14:10:55 -07:00 committed by TensorFlower Gardener
parent fb6e0c1cdd
commit 67edd1327a
1 changed files with 24 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
tensorflow/compiler/xla/g3doc/operation_semantics.md
View File

@ -811,15 +811,15 @@ Here is pseudo-code for a 2d convolution with padding and striding:
``` ```
for (b, oz, oy, ox) { // output coordinates for (b, oz, oy, ox) { // output coordinates
value = 0; value = 0;
for (iz, ky, kx) { // kernel coordinates and input z for (iz, ky, kx) { // kernel coordinates and input z
iy = oy*stride_y + ky - pad_low_y; iy = oy*stride_y + ky - pad_low_y;
ix = ox*stride_x + kx - pad_low_x; ix = ox*stride_x + kx - pad_low_x;
if ((iy, ix) inside the base area considered without padding) { if ((iy, ix) inside the base area considered without padding) {
value += input(b, iz, iy, ix) * kernel(oz, iz, ky, kx); value += input(b, iz, iy, ix) * kernel(oz, iz, ky, kx);
} }
} }
output(b, oz, oy, ox) = value; output(b, oz, oy, ox) = value;
} }
``` ```
@ -899,19 +899,19 @@ Here is an example of an implementation of `myfunc`:
``` ```
extern "C" void myfunc(void* out, void** in) { extern "C" void myfunc(void* out, void** in) {
float (&x)[2] = *static_cast<float(*)[2]>(in[0]); float (&x)[2] = *static_cast<float(*)[2]>(in[0]);
float (&y)[2][3] = *static_cast<float(*)[2][3]>(in[1]); float (&y)[2][3] = *static_cast<float(*)[2][3]>(in[1]);
EXPECT_EQ(1, x[0]); EXPECT_EQ(1, x[0]);
EXPECT_EQ(2, x[1]); EXPECT_EQ(2, x[1]);
EXPECT_EQ(10, y[0][0]); EXPECT_EQ(10, y[0][0]);
EXPECT_EQ(20, y[0][1]); EXPECT_EQ(20, y[0][1]);
EXPECT_EQ(30, y[0][2]); EXPECT_EQ(30, y[0][2]);
EXPECT_EQ(40, y[1][0]); EXPECT_EQ(40, y[1][0]);
EXPECT_EQ(50, y[1][1]); EXPECT_EQ(50, y[1][1]);
EXPECT_EQ(60, y[1][2]); EXPECT_EQ(60, y[1][2]);
float (&z)[3][3] = *static_cast<float(*)[3][3]>(out); float (&z)[3][3] = *static_cast<float(*)[3][3]>(out);
z[0][0] = x[1] + y[1][0]; z[0][0] = x[1] + y[1][0];
// ... // ...
} }
``` ```
@ -1694,11 +1694,11 @@ dependency between the while loops.
``` ```
result1 = while (condition, init = init_value) { result1 = while (condition, init = init_value) {
Infeed(shape) Infeed(shape)
} }
result2 = while (condition, init = result1) { result2 = while (condition, init = result1) {
Infeed(shape) Infeed(shape)
} }
``` ```