Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for Dtype (0.05 sec)

  1. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

            absl::StrCat(src_node->name(), "_", src_slot, "_retval"), kRetValOp,
            NodeDebugInfo(src_node->def()));
        DataType dtype = src_node->output_type(src_slot);
        builder.Attr("T", dtype);
        builder.Attr("index", ret_index);
        builder.Input(src_image->name(), src_slot, dtype);
        Status s = builder.Finalize(&ret_def);
        if (!s.ok()) return s;
        TF_ASSIGN_OR_RETURN(Node * ret, graph_->AddNode(ret_def));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

          OpT op, typename OpT::Adaptor adaptor,
          ConversionPatternRewriter &rewriter) const override {
        Type dtype = op.getElementDtype();
        if (!(dtype.isF16() || dtype.isF32() || dtype.isF64() ||
              dtype.isInteger(1) || dtype.isInteger(8) || dtype.isInteger(16) ||
              dtype.isInteger(32) || dtype.isInteger(64))) {
          const char *error_info =
              "requires element_dtype to be 1-bit/8-bit/16-bit/32-bit/64-bit "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

                                             .WithAttr("dtype", DT_FLOAT)
                                             .WithAttr("value", Tensor()));
        Node* b = ops::SourceOp("Const", builder.opts()
                                             .WithName("b")
                                             .WithAttr("dtype", DT_FLOAT)
                                             .WithAttr("value", Tensor()));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_test.cc

      TF_OperationDescription* init(string type) {
        // Construct op_name to match the name used by REGISTER_OP in the
        // ATTR_TEST_REGISTER calls above.
        string op_name = "CApiAttributesTestOp";
        if (type.find("list(") == 0) {
          op_name += "List";
          type = type.replace(0, 5, "");
          type = type.replace(type.size() - 1, 1, "");
        }
        op_name += type;
        return TF_NewOperation(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc

        LLVM_DEBUG(llvm::dbgs() << "Zero point value should be a tensor type. Got: "
                                << zero_points_type << ".\n");
        return failure();
      }
    
      if (Type zero_points_element_type = zero_points_type.getElementType();
          !mlir::isa<IntegerType>(zero_points_element_type)) {
        LLVM_DEBUG(llvm::dbgs() << "Zero point should be an integer type. Got: "
                                << zero_points_element_type << ".\n");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        return false;
      }
    
      if (n->type_string() != "AssignAddVariableOp" &&
          n->type_string() != "AssignSubVariableOp") {
        return false;
      }
    
      DataType dtype;
      if (!TryGetNodeAttr(n->def(), "dtype", &dtype) || !DataTypeIsInteger(dtype)) {
        return false;
      }
    
      Node* const_input = nullptr;
      for (const Edge* e : n->in_edges()) {
        if (!e->IsControlEdge() && e->src()->IsConstant()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

    }
    
    // Return true if the passed quantized type is unsigned.
    bool QuantizedTypeIsUnsigned(Type type) {
      return TypeSwitch<Type, bool>(type)
          .Case<mlir::TF::Qint8Type>([](Type) { return false; })
          .Case<mlir::TF::Qint16Type>([](Type) { return false; })
          .Case<mlir::TF::Qint32Type>([](Type) { return false; })
          .Case<mlir::TF::Quint8Type>([](Type) { return true; })
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/deadness_analysis.cc

        if (node->type_string() == "Const" && must_be_true) {
          const TensorProto* proto = nullptr;
          TF_RETURN_IF_ERROR(GetNodeAttr(node->def(), "value", &proto));
    
          Tensor tensor(proto->dtype());
          TF_RET_CHECK(tensor.FromProto(*proto));
    
          *predicate = tensor.scalar<bool>()() ? MakeTrue() : MakeFalse();
          return absl::OkStatus();
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 60.4K bytes
    - Viewed (0)
Back to top