Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for check_numerics (0.23 sec)

  1. tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir

      // CHECK:  return %arg0 : tensor<3xi32>
    }
    
    func.func @CheckNumerics(%arg0: tensor<3xf32>) -> tensor<3xf32> {
      %0 = "tf.CheckNumerics"(%arg0) {message = ""}: (tensor<3xf32>) -> tensor<3xf32>
      func.return %0 : tensor<3xf32>
      // Should be converted to Identity and then from Identity to value
      // CHECK-LABEL: CheckNumerics
      // CHECK:  return %arg0 : tensor<3xf32>
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 59.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

      Scope root = Scope::NewRootScope().ExitOnError();
      Output a = ops::Placeholder(root.WithOpName("test/a"), DT_FLOAT);
      Output b = ops::Placeholder(root.WithOpName("test/b"), DT_FLOAT);
      Output check =
          ops::CheckNumerics(root.WithOpName("test/check"), a, "test/check");
      Output ge = ops::GreaterEqual(root.WithOpName("test/greaterequal"), check, b);
      Operation assert = ops::Assert(root.WithOpName("test/assert"), ge, {a, b});
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/mark_for_compilation_pass.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top