Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ApplyXlaHostTransferAttr (0.15 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc 

      if (device_ordinal)
    return ApplyXlaHostTransferAttr(
        builder.create<mlir::TF::_XlaSendFromHostV2Op>(
            loc, inputs,
            /*dynamic_key=*/compilation_key, device_ordinal,
            builder.getStringAttr(communication_key), device_type_attr),
        builder);

  return ApplyXlaHostTransferAttr(
      builder.create<mlir::TF::_XlaSendFromHostOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
Back to top