- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for Dtype (0.05 sec)
-
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) -
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) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
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) -
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) -
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) -
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) -
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)