Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for called_index (0.21 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc

    // i64 attribute `called_index` to the TF function's name.
    constexpr llvm::StringRef kTfBackendConfigAttrName = "tf.backend_config";
    constexpr llvm::StringRef kCalledIndexAttrName = "called_index";
    constexpr llvm::StringRef kCalledFuncAttrName = "called_func";
    
    // Converts `called_func` attributes in custom call ops back to `called_index`.
    FailureOr<ArrayAttr> DesymbolizeCustomCallCalledIndex(ModuleOp module) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/xla_call_module_deserialization.mlir

        // CHECK:     return %[[RESULT]]
        func.return %0 : tensor<10xi32>
      }
    
      // CHECK-LABEL: func @foo
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 02 18:38:51 UTC 2023
    - 6.9K bytes
    - Viewed (0)
Back to top