Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 46 of 46 for Dtype (0.77 sec)

  1. tensorflow/compiler/mlir/lite/utils/constant_utils.cc

        } else {
          return tensorflow::Status(absl::StatusCode::kInvalidArgument,
                                    "Unsupported type");
        }
      } else if (auto itype = mlir::dyn_cast<mlir::IntegerType>(element_type)) {
        if (element_type.isSignedInteger()) {
          switch (itype.getWidth()) {
            case 8:
              return DenseElementsAttr::get<int8_t>(shaped_type,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.5K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/ops/gen/cpp/views/attr_view.cc

    string AttrView::DefaultValue() const {
      const AttrValue &attr_value = attr_.default_value();
      switch (attr_value.value_case()) {
        case AttrValue::VALUE_NOT_SET:
          return "";
        case AttrValue::kType:
          return DataType_Name(attr_value.type());
        case AttrValue::kS:
          return "\"" + attr_value.s() + "\"";
        case AttrValue::kI:
          return std::to_string(attr_value.i());
        case AttrValue::kF:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 03 07:02:00 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc

    func::FuncOp BuildGELUDecomposition(RankedTensorType type,
                                        PatternRewriter& rewriter,
                                        Block* insertion_point) {
      rewriter.setInsertionPointToStart(insertion_point);
    
      auto ftype = FunctionType::get(rewriter.getContext(), {type}, {type});
      auto name = rewriter.getStringAttr("gelu_decomp");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 9.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/utils/convert_type.cc

    tflite::TensorType ConvertTypeToTensorType(mlir::Type type) {
      if (type.isF16()) {
        return tflite::TensorType_FLOAT16;
      } else if (type.isBF16()) {
        return tflite::TensorType_BFLOAT16;
      } else if (type.isF32()) {
        return tflite::TensorType_FLOAT32;
      } else if (type.isF64()) {
        return tflite::TensorType_FLOAT64;
      } else if (mlir::isa<mlir::TF::StringType>(type)) {
        return tflite::TensorType_STRING;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc

      return success();
    }
    
    static LogicalResult FoldDivOp(stablehlo::DivOp op, PatternRewriter& rewriter) {
      auto etype = op.getType().getElementType();
      if (etype.isa<FloatType>()) {
        return FoldDivOpInternal<APFloat>(op, rewriter);
      }
      if (etype.isa<IntegerType>()) {
        return FoldDivOpInternal<APInt>(op, rewriter);
      }
      return failure();
    }
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 06:11:55 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc

              auto attr_key = attr.first;
              auto attr_value = attr.second;
              if (attr_key == attrKey &&
                  attr_value.value_case() == AttrValue::kType &&
                  attr_value.type() == DT_RESOURCE) {
                return true;
                break;
              }
            }
          }
        }
      }
      return false;
    }
    
    bool IsNonReplicatedGraph(const Graph& graph,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 12:22:33 UTC 2024
    - 8.9K bytes
    - Viewed (0)
Back to top