Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for shape (0.03 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top