Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for CreateSendOp (0.09 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc

          ArrayRef<NamedAttribute>{rendezvous_name_attr, host_handler_name_attr});
      op->setAttr(kFrontendAttributesAttr, frontend_attributes);
    }
    
    // Creates a `mhlo.send` op for sending value `operand`.
    Value CreateSendOp(OpBuilder& builder, Location loc, Value operand,
                       StringRef key, size_t index, Value token,
                       StringRef host_handler_name, bool manual_sharding) {
      // type 2 == DEVICE_TO_HOST
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 40.5K bytes
    - Viewed (0)
Back to top