- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for kXlaOutsideCompilationAttr (0.32 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/tensorflow/transforms/tpu_host_computation_expansion.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" namespace mlir { namespace TFTPU { namespace { constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; bool HasOutsideCompilationAttribute(Operation* op) { return op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; } // Finds op that created a given value. If the value is a BlockArgument, this // returns the owner of the Block.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/verify_no_outside_compilation_markers_pass.cc
#include "mlir/Support/LLVM.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_device.h" namespace mlir { namespace TFDevice { namespace { constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; #define GEN_PASS_DEF_VERIFYNOOUTSIDECOMPILATIONMARKERSPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 22 19:52:08 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_update_embedding_enqueue_op_inputs.cc
mode_string_value)); auto outside_compilation_attr = embedding_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr); if (outside_compilation_attr) enqueue_mode->setAttr(kXlaOutsideCompilationAttr, outside_compilation_attr); mode_enqueue_operand.set(enqueue_mode); } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.7K 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/tf2xla/internal/passes/verify_clustering_pass.cc
" not in tf functional dialect"; op->emitError() << error; return WalkResult::interrupt(); } if (op->hasAttr(mlir::TF::kXlaOutsideCompilationAttr)) { std::string error = "op has outside compilation attribute _xla_outside_compilation which " "is not allowed after clustering"; op->emitError() << error;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 20:33:37 UTC 2023 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
if (walked_op->hasAttr(kXlaOutsideCompilationAttr)) { return WalkResult::interrupt(); } return WalkResult::advance(); }) .wasInterrupted(); } // Returns whether `op` or any ancestors of `op` are outside compiled. bool HasOutsideCompilationAncestor(Operation* op) { while (op) { if (op->hasAttr(kXlaOutsideCompilationAttr)) { return true; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h
inline constexpr llvm::StringRef kTpuReplicateAttr = "_tpu_replicate"; // Device types. inline constexpr llvm::StringRef kTpuDevice = "TPU"; // _xla_outside_compilation inline constexpr llvm::StringRef kXlaOutsideCompilationAttr = "_xla_outside_compilation"; // device attr inline constexpr llvm::StringRef kDeviceAttr = "device"; // Function attribute to signal that a function should be skipped from TPU
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 8.7K 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)