- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for recv_device (0.15 sec)
-
tensorflow/c/eager/c_api_test_util.cc
send_device.size()); TFE_OpSetAttrString(op, "recv_device", recv_device.c_str(), 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) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.h
// Return a SendOp op `op_name` with send input tensor `in` and attributes // `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);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 17 23:43:59 UTC 2023 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_remaining_ops.cc
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 std::optional<std::string> _HostRecvOp::GetResourceInstanceStr() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 20:05:58 UTC 2023 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tpu_conversions.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 04 21:25:31 UTC 2023 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
} def TF_RecvOp : TF_Op<"Recv", [TF_RecvSideEffect]> { 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
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)