Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for send_device_incarnation (0.54 sec)

  1. tensorflow/c/eager/c_api_test_util.h

    // `send_device`, `recv_device`, and `send_device_incarnation` set.
    TFE_Op* SendOp(TFE_Context* ctx, TFE_TensorHandle* in,
                   const std::string& op_name, const std::string& send_device,
                   const std::string& recv_device,
                   tensorflow::uint64 send_device_incarnation);
    
    // Return a RecvOp op `op_name` with the attributes `send_device`,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jul 17 23:43:59 UTC 2023
    - 7.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_test_util.cc

                          recv_device.size());
      TFE_OpSetAttrInt(op, "send_device_incarnation", send_device_incarnation);
    
      return op;
    }
    
    TFE_Op* RecvOp(TFE_Context* ctx, const std::string& op_name,
                   const std::string& send_device, const std::string& recv_device,
                   tensorflow::uint64 send_device_incarnation) {
      TF_Status* status = TF_NewStatus();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_remaining_ops.cc

    namespace {
    std::string GetRendezvousKey(const std::string& send_device,
                                 const uint64_t send_device_incarnation,
                                 const std::string& recv_device,
                                 const std::string& tensor_name) {
      return absl::StrCat(send_device, ";", send_device_incarnation, ";",
                          recv_device, ";", tensor_name);
    }
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 14 20:05:58 UTC 2023
    - 7.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfrt/tests/mlrt/tpu_conversions.mlir

      // CHECK: tf_mlrt.await [[result_future]]
      %0 = "tf.Recv"() {__op_key = 0: i32, device = "/device:CPU:0", tensor_name = "tensor", send_device = "/device:CPU:0", send_device_incarnation = 0, recv_device = "/device:CPU:0"} : () -> tensor<f32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 04 21:25:31 UTC 2023
    - 11K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      let summary = "Receives the named tensor from send_device on recv_device.";
    
      let arguments = (ins
        StrAttr:$tensor_name,
        StrAttr:$send_device,
        I64Attr:$send_device_incarnation,
        StrAttr:$recv_device,
        DefaultValuedOptionalAttr<BoolAttr, "false">:$client_terminated
      );
    
      let results = (outs
        Res<TF_Tensor, [{The tensor to receive.}]>:$tensor
      );
    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