Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for getAttrOfType (0.23 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        auto bias =
            rewriter.create<TF::ConstOp>(op->getLoc(), bias_type, bias_attr);
    
        if (op->getAttrOfType<StringAttr>("padding").getValue() == "EXPLICIT") {
          // Add Const op for padding value.
          ArrayRef<Attribute> padding_attr_array =
              op->getAttrOfType<ArrayAttr>("explicit_paddings").getValue();
    
          auto get_int = [](Attribute attr) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

        // contain the attribute names.
        // TransferQuantizationAttributes(rewriter, inner_op, attrs);
        std::string attr_map_str =
            inner_op.getAttrOfType<StringAttr>(kAttrMapAttribute).str();
        for (absl::string_view element_str : absl::StrSplit(attr_map_str, ',')) {
          std::vector<absl::string_view> key_and_value_pair =
              absl::StrSplit(element_str, ':');
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

    // outside compilation.
    LogicalResult CheckAncestorNotOutsideComp(Operation* op) {
      if (!op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr))
        return mlir::success();
      Operation* iter_op = op;
      while (auto* parent_op = iter_op->getParentOp()) {
        if (parent_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) {
          return op->emitOpError()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

      return true;
    }
    
    StringAttr GetReplicationAttr(mlir::Operation* op) {
      return op->getAttrOfType<StringAttr>(TF::kReplicationInfoAttr);
    }
    
    StringAttr GetReplicationAttr(TF::TPUCompilationResultOp op) {
      // Special case for getting the replication region for
      // TPUCompilationResultsOp.
      return op->getAttrOfType<StringAttr>(kTpuCompilationStatus);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

        const std::vector<std::unique_ptr<tflite::TensorT>>& tensors) {
      auto* context = func->getContext();
      static const char kEntryFunctionAttributes[] = "tf.entry_function";
    
      auto dict_attr =
          func->getAttrOfType<mlir::DictionaryAttr>(kEntryFunctionAttributes);
      if (!dict_attr) return;
    
      // Get Input and output tensor names from attribute.
      llvm::SmallVector<llvm::StringRef, 2> input_names =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
Back to top