- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 125 for attr_name (0.55 sec)
-
tensorflow/c/python_api.cc
} void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, TF_Buffer* attr_value_proto, TF_Status* status) { TF_SetAttr(graph, op, attr_name, attr_value_proto, status); } void ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, TF_Status* status) { TF_ClearAttr(graph, op, attr_name, status); } void SetFullType(TF_Graph* graph, TF_Operation* op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 12 18:48:56 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc
return true; } bool GetI32VectorFromDenseI64CompositeAttr( const DictionaryAttr& composite_attrs, const std::string& attr_name, std::vector<int32_t>* out_vec) { DenseIntElementsAttr attr; if (!EnsureAttribute<DenseIntElementsAttr>(composite_attrs, attr_name, &attr)) { return false; } return DenseI64AttrToI32Vector(attr, out_vec); }
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/c/kernels.h
TF_CAPI_EXPORT extern TF_Buffer* TF_OpKernelConstruction_GetAttrFunction( TF_OpKernelConstruction* ctx, const char* attr_name, TF_Status* status); // Return true if the kernel construction has the attr_name TF_CAPI_EXPORT extern bool TF_OpKernelConstruction_HasAttr( TF_OpKernelConstruction* ctx, const char* attr_name, TF_Status* status); // Returns the unique operation name for this OpKernel.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:46:22 UTC 2024 - 24.6K bytes - Viewed (0) -
tensorflow/c/kernels.cc
case DataTypeToEnum<type>::value: { \ kernel_builder->cc_builder->TypeConstraint<type>(attr_name); \ break; \ } void AddTypeConstraint(TF_KernelBuilder* kernel_builder, const char* attr_name, const DataType dtype, TF_Status* status) { // This needs to be under tensorflow:: namespace so that
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/c/c_api.cc
} desc->node_builder.Attr(attr_name, v); } } void TF_SetAttrInt(TF_OperationDescription* desc, const char* attr_name, int64_t value) { desc->node_builder.Attr(attr_name, static_cast<int64_t>(value)); } void TF_SetAttrIntList(TF_OperationDescription* desc, const char* attr_name, const int64_t* values, int num_values) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.h
TF_CAPI_EXPORT extern void TFE_OpSetAttrInt(TFE_Op* op, const char* attr_name, int64_t value); TF_CAPI_EXPORT extern void TFE_OpSetAttrFloat(TFE_Op* op, const char* attr_name, float value); TF_CAPI_EXPORT extern void TFE_OpSetAttrBool(TFE_Op* op, const char* attr_name, unsigned char value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
const std::string& attr_name, int N) { ArrayAttr array_attr = mlir::dyn_cast_or_null<ArrayAttr>(attrs.get(attr_name)); if (array_attr == nullptr || array_attr.size() != N) { return func->emitWarning() << "'" << attr_name << "' attribute for " << kMaxUnpooling << " must be set and has size of " << N;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/c/c_api.h
int64_t total_size; } TF_AttrMetadata; // Returns metadata about the value of the attribute `attr_name` of `oper`. TF_CAPI_EXPORT extern TF_AttrMetadata TF_OperationGetAttrMetadata( TF_Operation* oper, const char* attr_name, TF_Status* status); // Fills in `value` with the value of the attribute `attr_name`. `value` must // point to an array of length at least `max_length` (ideally set to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 26 21:08:15 UTC 2023 - 82.3K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
if (!s.ok() && s.code() != error::NOT_FOUND) { return s; } n->ClearAttr(attr_name); attr_value.push_back(value); n->AddAttr(attr_name, attr_value); return absl::OkStatus(); } // Replaces attribute value. template <typename T> void ReplaceAttr(Node* n, const string& attr_name, const T& value) { n->ClearAttr(attr_name); n->AddAttr(attr_name, value); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
for (auto& attr : func->getAttrs()) { StringRef attr_name = attr.getName().getValue(); if (attr_name.starts_with("tf_saved_model.")) { func->removeAttr(attr_name); } } auto iface = cast<FunctionOpInterface>(func.getOperation()); for (int i = 0; i < func.getNumArguments(); ++i) { for (auto& attr : iface.getArgAttrs(i)) { const StringAttr& attr_name = attr.getName();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0)