Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for XlaCustomCallV2 (0.24 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/jit/mark_for_compilation_pass.cc 

          "While",
      "XlaAllReduce",
      "XlaBroadcastHelper",
      "XlaCallModule",
      "XlaConcatND",
      "XlaConv",
      "XlaConvV2",
      "XlaCustomCall",
      "XlaCustomCallV2",
      "XlaDequantize",
      "XlaDot",
      "XlaDotV2",
      "XlaDynamicSlice",
      "XlaDynamicUpdateSlice",
      "XlaEinsum",
      "XlaGather",
      "XlaIf",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  3. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td 

      TF_DerivedResultTypeAttr preferred_element_type = TF_DerivedResultTypeAttr<0>;

  let hasVerifier = 1;
}

def TF_XlaCustomCallV2Op : TF_Op<"XlaCustomCallV2", [Pure]> {
  let summary = "Emits an HLO `CustomCall` operation with multiple outputs.";

  let description = [{
As opposed to `XlaCustomCall`, this operation supports multiple outputs.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top