- Sort Score
- Result 10 results
- Languages All
Results 91 - 100 of 126 for attrName (0.19 sec)
-
tensorflow/c/ops.cc
TF_ShapeInferenceContext* ctx, const char* attr_name, c_type* val, \ TF_Status* status) { \ TF_SetStatus(status, TF_OK, ""); \ cc_type v; \ auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); \ Status s = cc_ctx->GetAttr(attr_name, &v); \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 28 22:41:35 UTC 2022 - 10.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.h
// `op_name` must outlive `op`. void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name, TF_Status* s); // `attr_name` must outlive `op`. void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name, TF_DataType value, TF_Status* s); void TF_DeleteAbstractTensor(TF_AbstractTensor*);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Oct 24 11:27:00 UTC 2021 - 7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
} tsl::Set_TF_Status_from_Status(s, tracing_op->SetOpName(op_name)); } void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name, TF_DataType value, TF_Status* s) { Status status = unwrap(op)->SetAttrType(attr_name, static_cast<DataType>(value)); TF_SetStatus(s, static_cast<TF_Code>(status.code()), absl::StatusMessageAsCStr(status)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 10:15:17 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
} else { auto attr_name = compose_func.getArgAttrOfType<StringAttr>( arg.index(), kAttrArgumentNameAttr); auto attribute = op->getAttr(attr_name.getValue()); if (!attribute) { attribute = compose_func.getArgAttr(arg.index(), kAttrArgumentDefaultAttr); } if (!attribute && attr_name.getValue() == "out_type") {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
void TF_AttrBuilderSetType(TF_AttrBuilder* builder, const char* attr_name, TF_DataType value) { auto iter = builder->attr_names.insert(attr_name).first; builder->Set(*iter, static_cast<tensorflow::DataType>(value)); } void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder, const char* attr_name, const TF_DataType* values, int num_values) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
if (operand.index() >= call_op.getNumOperands()) { auto attr_name = signature.getArgAttrOfType<StringAttr>( operand.index(), kAttrArgumentNameAttr); auto attr_value = signature.getArgAttr(operand.index(), kAttrArgumentDefaultAttr); arg_attrs->push_back( rewriter.getNamedAttr(attr_name.getValue(), attr_value)); continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/c/kernels_experimental.h
// array of length at least `max_values` (ideally set to total_size from // TF_OpKernelConstruction_GetAttrSize(ctx, attr_name, &list_size, // &total_size)). TF_CAPI_EXPORT extern void TF_OpKernelConstruction_GetAttrTensorShape( TF_OpKernelConstruction* ctx, const char* attr_name, int64_t* dims, size_t num_dims, TF_Status* status); TF_CAPI_EXPORT extern bool TF_IsRefInput(TF_OpKernelContext* ctx, int i,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 07 14:44:39 UTC 2023 - 9.4K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_util.cc
const bool use_const = entry.second; string attr_name = AvoidCPPKeywords(api_def_attr.rename_to()); string attr_comment; if (!api_def_attr.description().empty()) { // TODO(keveman): Word wrap and indent this, to handle multi-line // descriptions. strings::StrAppend(&attr_comment, "* ", attr_name, ": ", api_def_attr.description(), "\n"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 25K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
} } auto check_i64_attr = [&](const std::string& attr_name) -> mlir::LogicalResult { if (!backend_config.contains(attr_name)) { return op.emitOpError() << "Missing " << attr_name << " attribute in backend_config"; } auto attr = backend_config.getAs<IntegerAttr>(attr_name); if (!attr || !attr.getType().isInteger(64)) { return op.emitOpError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
auto sizes = mlir::DenseI32ArrayAttr::get(context, {1, 1, 1, 0}); auto attr_name = StringAttr::get(context, mlir::OpTrait::AttrSizedOperandSegments< void>::getOperandSegmentSizeAttr()); attrs.push_back(NamedAttribute(attr_name, sizes)); } // Insert fused operation right before the BiasAdd operation to guarantee
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0)