Skip to content

Commit

Permalink
fix build break
Browse files Browse the repository at this point in the history
  • Loading branch information
yufenglee committed Jan 27, 2022
1 parent b5e4e24 commit 119e753
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion include/onnxruntime/core/graph/graph.h
Original file line number Diff line number Diff line change
Expand Up @@ -357,10 +357,10 @@ class Node {
/** Gets the Node's attributes. */
const NodeAttributes& GetAttributes() const noexcept { return attributes_; }

#if !defined(ORT_MINIMAL_BUILD)
/** Remove the specified attribute from this Node */
bool ClearAttribute(const std::string& attr_name);

#if !defined(ORT_MINIMAL_BUILD)
/** Gets the Node's mutable attributes. */
NodeAttributes& GetMutableAttributes() noexcept { return attributes_; }

Expand Down
2 changes: 1 addition & 1 deletion onnxruntime/core/graph/graph.cc
Original file line number Diff line number Diff line change
Expand Up @@ -938,13 +938,13 @@ ADD_ATTR_IMPL(SparseTensorProto, AttributeProto_AttributeType::AttributeProto_At
ADD_LIST_ATTR_IMPL(SparseTensorProto, AttributeProto_AttributeType::AttributeProto_AttributeType_SPARSE_TENSORS, sparse_tensors)
#endif

#if !defined(ORT_MINIMAL_BUILD)
bool Node::ClearAttribute(const std::string& attr_name) {
graph_->SetGraphResolveNeeded();
graph_->SetGraphProtoSyncNeeded();
return attributes_.erase(attr_name) > 0;
}

#if !defined(ORT_MINIMAL_BUILD)
Status Node::UpdateInputArgCount() {
// The node refers to a primitive operator.
// Infer and verify node input arg type information.
Expand Down
2 changes: 1 addition & 1 deletion onnxruntime/test/optimizer/qdq_transformer_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -640,7 +640,7 @@ void QDQTransformerGemmTests(bool has_output_q, bool has_bias, bool beta_not_one
if (has_bias) {
auto* dq_bias_output = builder.MakeIntermediate();
auto* bias = builder.MakeInitializer<BiasType>({input2_shape[1]}, static_cast<BiasType>(0), static_cast<BiasType>(127));
builder.AddDequantizeLinearNode<BiasType>(bias, 0.00156,
builder.AddDequantizeLinearNode<BiasType>(bias, 0.00156f,
0,
dq_bias_output);
input_args.push_back(dq_bias_output);
Expand Down

0 comments on commit 119e753

Please sign in to comment.