- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 125 for attr_name (0.47 sec)
-
tensorflow/c/eager/abstract_operation.h
virtual Status SetAttrString(const char* attr_name, const char* data, size_t length) = 0; virtual Status SetAttrInt(const char* attr_name, int64_t value) = 0; virtual Status SetAttrFloat(const char* attr_name, float value) = 0; virtual Status SetAttrBool(const char* attr_name, bool value) = 0; virtual Status SetAttrType(const char* attr_name, DataType value) = 0;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 14 16:20:41 UTC 2021 - 6.8K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
return op_->SetAttrInt(attr_name, value); } Status SetAttrFloat(AbstractOperation* op_, const char* attr_name, float value, ForwardOperation* forward_op_) { forward_op_->attrs.Set(attr_name, value); return op_->SetAttrFloat(attr_name, value); } Status SetAttrBool(AbstractOperation* op_, const char* attr_name, bool value,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
op_->node_builder.Attr(attr_name, value); return absl::OkStatus(); } Status SetAttrBool(const char* attr_name, bool value) override { op_->node_builder.Attr(attr_name, value); return absl::OkStatus(); } Status SetAttrType(const char* const attr_name, DataType value) override { if (!op_) { return Status(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 20:00:09 UTC 2024 - 15.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
} } void TFE_OpSetAttrInt(TFE_Op* op, const char* attr_name, int64_t value) { auto s = tensorflow::unwrap(op)->SetAttrInt(attr_name, value); if (!s.ok()) { LOG(WARNING) << "Unable to set attribute: " << attr_name; } } void TFE_OpSetAttrFloat(TFE_Op* op, const char* attr_name, float value) { auto s = tensorflow::unwrap(op)->SetAttrFloat(attr_name, value); if (!s.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
forward_op_.attrs.Set(attr_name, value); return parent_op_->SetAttrFloat(attr_name, value); } Status TapeOperation::SetAttrBool(const char* attr_name, bool value) { forward_op_.attrs.Set(attr_name, value); return parent_op_->SetAttrBool(attr_name, value); } Status TapeOperation::SetAttrType(const char* attr_name, DataType value) { forward_op_.attrs.Set(attr_name, value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 06:16:45 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.td
// with the key `attr_name` as the type provided by `attr_type`. class GetCompositeAttributeAs<string attr_name, string attr_type>: NativeCodeCall<"$0.get(\"" # attr_name # "\").dyn_cast<" # attr_type # ">()">; // Receives a composite DictionaryAttr and returns the value of the Attribute // with the key `attr_name` as a DenseIntElementsAttr. class GetAsVectorAttr<string attr_name>:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:05:30 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/utils/utils.cc
for (auto& attr : src->getAttrs()) { StringRef attr_name = attr.getName().strref(); if (!registered.contains(attr_name) && !(GetAllowedAttributes().contains(attr_name) || GetOptionalAttributes().contains(attr_name))) { src->emitError("Denied unregistered attribute was found: " + attr_name); return failure(); } } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 6.2K bytes - Viewed (0) -
tensorflow/c/eager/gradients_internal.h
Status SetAttrString(AbstractOperation*, const char* attr_name, const char* data, size_t length, ForwardOperation*); Status SetAttrInt(AbstractOperation*, const char* attr_name, int64_t value, ForwardOperation*); Status SetAttrFloat(AbstractOperation*, const char* attr_name, float value, ForwardOperation*); Status SetAttrBool(AbstractOperation*, const char* attr_name, bool value,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Oct 24 11:27:35 UTC 2021 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.h
namespace mlir { namespace odml { // Ensure an attribute named attr_name exists and it is of type AttrType. // If so, sets the `out_attr` pointer to point to the casted attribute. template <typename AttrType> bool EnsureAttribute(const DictionaryAttr& composite_attributes, const std::string& attr_name, AttrType* out_attr) { Attribute attr = composite_attributes.get(attr_name); if (!mlir::isa_and_nonnull<AttrType>(attr)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 18:33:05 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
Status SetAttrString(const char* attr_name, const char* data, size_t length) override; Status SetAttrInt(const char* attr_name, int64_t value) override; Status SetAttrFloat(const char* attr_name, float value) override; Status SetAttrBool(const char* attr_name, bool value) override; Status SetAttrType(const char* attr_name, tensorflow::DataType dtype) override;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0)