- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for kXlaOutsideCompilationAttr (0.37 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
!HasOutsideCompiledAncestor(input_defining_op) && !input_defining_op->hasAttrOfType<StringAttr>( kXlaOutsideCompilationAttr)) { input_defining_op->setAttr( kXlaOutsideCompilationAttr, StringAttr::get(input_defining_op->getContext(), "auto")); outside_compiled_ops.push(input_defining_op); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
using mlir::StringAttr; using mlir::Type; using mlir::Value; using mlir::WalkResult; using mlir::func::FuncOp; constexpr char kXlaMapOutsideCompilationAttr[] = "_xla_map_outside_compilation"; constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; // Return true if `op` has attributes that say it can be outside compiled by // this pass. This pass ignores _xla_map_outside_compilation, which will only be
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc
if (metadata_map.find(cluster) == metadata_map.end()) return false; auto metadata = metadata_map[cluster]; if (!metadata.getAllowSoftPlacement() && !op->hasAttr(TF::kXlaOutsideCompilationAttr)) return true; std::string device = ""; if (op->hasAttr(TF::kDeviceAttr)) device = op->getAttrOfType<StringAttr>(TF::kDeviceAttr).str(); else return false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 21.5K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
<< node.type_string() << ")" << (reason.empty() ? "" : ": ") << reason; } bool IsInOutsideCompilationCluster(const Node& n) { return n.attrs().Find(kXlaOutsideCompilationAttr) != nullptr; } Status MakeCallNodeFromAttribute(const Node& node, const std::string& attr_name, NodeDef* node_def) { const NameAttrList* name_attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0)