- Sort Score
- Result 10 results
- Languages All
Results 71 - 73 of 73 for BoolAttr (0.14 sec)
-
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
} // Default to non-time_major. std::optional<mlir::NamedAttribute> time_major_attr = attr.getNamed("time_major"); bool time_majored = time_major_attr ? time_major_attr->getValue().cast<BoolAttr>().getValue() : false; int64_t batch = time_majored ? input_type.getDimSize(1) : input_type.getDimSize(0); int64_t time =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
auto composite_control_flow_op = [&](const std::string& name) { result.name = mlir::OperationName(get_full_op_name(name), context_); bool stateless = absl::StartsWith(node_type_name, "Stateless"); mlir::BoolAttr val = builder_.getBoolAttr(stateless); result.attributes.push_back(builder_.getNamedAttr("is_stateless", val)); }; // Map Case/If/While and StatelessCase/If/While op in TensorFlow to the common
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// If the 'transpose' attribute is true returns ElementsAttr to transpose 2D // matrix. Otherwise, returns ElementsAttr for identity transpose. static DenseIntElementsAttr Get2DTransposePerm(BoolAttr transpose, Builder *b) { if (transpose.getValue()) return GetI64ElementsAttr({1, 0}, b); return GetI64ElementsAttr({0, 1}, b); } //===----------------------------------------------------------------------===//
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)