Search Options

Results per page
Sort
Preferred Languages
Advance

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

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

              return WalkResult::interrupt();
            }
            int called_index = called_index_attr.getInt();
            if (called_index < 0 || called_index >= function_list.size()) {
              op->emitOpError()
                  << "references function #" << called_index
                  << " but enclosing XlaCallModule has a function list of size "
                  << function_list.size();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  2. 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)
  3. 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