You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
@@ -1412,7 +1412,7 @@ be set to the first input's shape. If the output is selected by its index as in
1412
1412
1413
1413
There are a number of common shape functions
1414
1414
that apply to many ops, such as `shape_inference::UnchangedShape` which can be
1415
-
found in [common_shape_fns.h](https://www.tensorflow.org/code/tensorflow/core/framework/common_shape_fns.h) and used as follows:
1415
+
found in [common_shape_fns.h](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/framework/common_shape_fns.h) and used as follows:
1416
1416
1417
1417
```c++
1418
1418
REGISTER_OP("ZeroOut")
@@ -1459,7 +1459,7 @@ provides access to the attributes of the op).
1459
1459
1460
1460
Since shape inference is an optional feature, and the shapes of tensors may vary
1461
1461
dynamically, shape functions must be robust to incomplete shape information for
1462
-
any of the inputs. The `Merge` method in [`InferenceContext`](https://www.tensorflow.org/code/tensorflow/core/framework/shape_inference.h)
1462
+
any of the inputs. The `Merge` method in [`InferenceContext`](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/framework/shape_inference.h)
1463
1463
allows the caller to assert that two shapes are the same, even if either
1464
1464
or both of them do not have complete information. Shape functions are defined
1465
1465
for all of the core TensorFlow ops and provide many different usage examples.
@@ -1484,7 +1484,7 @@ If you have a complicated shape function, you should consider adding a test for
1484
1484
validating that various input shape combinations produce the expected output
1485
1485
shape combinations. You can see examples of how to write these tests in some
Copy file name to clipboardexpand all lines: site/en/guide/dtensor_overview.ipynb
+2-2
Original file line number
Diff line number
Diff line change
@@ -281,7 +281,7 @@
281
281
"id": "Eyp_qOSyvieo"
282
282
},
283
283
"source": [
284
-
"<img src=\"https://www.tensorflow.org/images/dtensor/dtensor_layout_rank2.png\" alt=\"A tensorr with it's first axis sharded across mesh dimension 'y' and it's second axis sharded across mesh dimension 'x'\" class=\"no-filter\">\n"
284
+
"<img src=\"https://www.tensorflow.org/images/dtensor/dtensor_layout_rank2.png\" alt=\"A tensorr with its first axis sharded across mesh dimension 'y' and it's second axis sharded across mesh dimension 'x'\" class=\"no-filter\">\n"
285
285
]
286
286
},
287
287
{
@@ -303,7 +303,7 @@
303
303
"source": [
304
304
"For the same `mesh_2d`, the layout `Layout([\"x\", dtensor.UNSHARDED], mesh_2d)` is a layout for a rank-2 `Tensor` that is replicated across `\"y\"`, and whose first axis is sharded on mesh dimension `x`.\n",
305
305
"\n",
306
-
"<img src=\"https://www.tensorflow.org/images/dtensor/dtensor_layout_hybrid.png\" alt=\"A tensor replicated across mesh-dimension y, with it's first axis sharded across mesh dimension 'x'\" class=\"no-filter\">\n"
306
+
"<img src=\"https://www.tensorflow.org/images/dtensor/dtensor_layout_hybrid.png\" alt=\"A tensor replicated across mesh-dimension y, with its first axis sharded across mesh dimension 'x'\" class=\"no-filter\">\n"
0 commit comments