Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for kXlaOutsideCompilationAttr (0.37 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
Back to top