- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 125 for attr_name (0.23 sec)
-
tensorflow/compiler/jit/compilability_check_util.h
UncompilableNodesMap* uncompilable_nodes) const; // Returns compilability of node def retrieved from `node`'s attribute with // name `attr_name`. bool ExtractNodeDefAndCheckCompilability( const Node& node, const std::string& attr_name, const std::string& call_name, NameAttrList* encapsulating_function, FunctionLibraryRuntime* lib_runtime, std::vector<StackFrameView>* stack_trace,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K 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) -
tensorflow/c/c_api_experimental.h
TF_CAPI_EXPORT extern void TF_AttrBuilderSetType(TF_AttrBuilder* builder, const char* attr_name, TF_DataType value); TF_CAPI_EXPORT extern void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder, const char* attr_name, const TF_DataType* values,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
} else if (n->IsIfNode()) { for (const string& attr_name : std::vector<string>{"then_branch", "else_branch"}) { NameAttrList branch_func; TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), attr_name, &branch_func)); (*branch_func.mutable_attr())["_device_ordinal"] = device_ordinal_value; n->ClearAttr(attr_name); n->AddAttr(attr_name, branch_func); } } else if (n->IsWhileNode()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen.cc
continue; } const string attr_name = api_def_attr.has_default_value() ? strings::StrCat("attrs.", api_def_attr.rename_to(), "_") : AvoidCPPKeywords(api_def_attr.rename_to()); strings::StrAppend(&body, spaces, ".Attr(\"", graph_attr.name(), "\", ", attr_name, ")\n"); } strings::StrAppend(&body, " ;\n");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 17:22:47 UTC 2023 - 17K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
all_attrs.insert(attr.getName().strref()); } for (const auto& operand : llvm::enumerate(getArgumentTypes())) { if (auto attr_name = getArgAttrOfType<StringAttr>( operand.index(), kAttrArgumentNameAttr)) { all_attrs.insert(attr_name.getValue()); } } return all_attrs; } }]; let hasVerifier = 1; let hasCustomAssemblyFormat = 1; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
TF_DeleteBuffer(buffer); TF_DeleteStatus(s); return ret; } bool GetAttrValue(TF_Operation* oper, const char* attr_name, tensorflow::AttrValue* attr_value, TF_Status* s) { TF_Buffer* buffer = TF_NewBuffer(); TF_OperationGetAttrValueProto(oper, attr_name, buffer, s); bool ret = TF_GetCode(s) == TF_OK; if (ret) ret = attr_value->ParseFromArray(buffer->data, buffer->length);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 15 03:16:52 UTC 2021 - 17.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
llvm::StringRef attr_name) { return attr_name.compare(kOutputShapesAttrName) == 0 && attr_value.value_case() == AttrValue::kList; } bool IsResourceOutputShapesAttribute(const AttrValue& attr_value, llvm::StringRef attr_name) { if (attr_name == "_handle_dtypes" || attr_name == "_handle_shapes")
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/c/kernels_experimental.cc
const char* attr_name, int64_t* dims, size_t num_dims, TF_Status* status) { ::tensorflow::TensorShape shape; auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); ::tensorflow::Status s = cc_ctx->GetAttr(attr_name, &shape); ::tensorflow::Set_TF_Status_from_Status(status, s);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0)