- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for _htod_0 (0.71 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
auto new_key = StringAttr::get(ctx, old_key.str() + "_dtoh_0"); op->setAttr("key", new_key); } else if (isa<TF::XlaRecvFromHostOp>(op)) { StringRef old_key = op->getAttrOfType<StringAttr>("key").getValue(); auto new_key = StringAttr::get(ctx, old_key.str() + "_htod_0"); op->setAttr("key", new_key); } }); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
MLIRContext* context = op->getContext(); std::string formatted_key = device_to_host ? llvm::formatv("{0}_dtoh_{1}", key, index).str() : llvm::formatv("{0}_htod_{1}", key, index).str(); auto rendezvous_name = StringAttr::get(context, formatted_key); auto rendezvous_name_attr = NamedAttribute( StringAttr::get(context, xla::kXlaHostTransferRendezvousNameAttr),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
Graph* g) { NodeDefBuilder send_pred_builder(name, "XlaSendToHost"); send_pred_builder.Attr("Tinput", DT_BOOL); send_pred_builder.Attr("key", absl::StrCat(host_transfer_key, "_dtoh_0")); send_pred_builder.Attr(kXlaTokenInputNodesAttrName, std::vector<string>{kXlaTokenArgNodeName}); send_pred_builder.Attr(kXlaOriginalOutsideCompilationNodeName, name);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)