- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for shape (0.04 sec)
-
tensorflow/cc/gradients/README.md
example: ``` TEST_F(ArrayGradTest, IdentityGrad) { TensorShape shape({5, 2}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape)); auto y = Identity(scope_, x); RunTest(x, shape, y, shape); } ``` NOTE: There are some ops that require features from the C++ API that are not yet implemented.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 29 19:12:55 UTC 2021 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/README.md
* `--tags`: Comma-separated list of tags for loading SavedModel. Ignored for MLIR input. * `--input-arg-shapes`: A string representation of input argument shapes for 'main' entry-point, separating tensors with ':', dimension with ',', and using '?' for unknown sizes. For example, `input-arg-shapes=1,2::1,?` expresses argument shapes `[1,2]`, `[]` and `[1,?]`. * `--e`: Elide large elements attrs while dumping the output StableHLO.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/README.md
new op is presented as a single node in the graph, thus optimization passes and kernels can easily be specialized to this op for better performance. * *Automatic shape/type inference support*: No shape functions are required for the new op; * *Automatic gradient support (WIP)*: The user doesn't need to author gradient a function of the op for training. ### Use Cases
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 29 18:32:13 UTC 2022 - 6.2K bytes - Viewed (0)