Searched refs:attr_tensor (Results 1 – 4 of 4) sorted by relevance
206 AttrValue attr_tensor; in FuseQuantizedConvolutionAndRequantize() local207 TensorProto* t = attr_tensor.mutable_tensor(); in FuseQuantizedConvolutionAndRequantize()209 bias_node->mutable_attr()->insert({"value", attr_tensor}); in FuseQuantizedConvolutionAndRequantize()
43 AttrValue attr_tensor; in AddNodeDivConst() local44 auto tensor = attr_tensor.mutable_tensor(); in AddNodeDivConst()47 node->mutable_attr()->insert({"value", attr_tensor}); in AddNodeDivConst()
417 AttrValue attr_tensor; in AddNodePermConst() local422 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodePermConst()423 node->mutable_attr()->insert({"value", attr_tensor}); in AddNodePermConst()443 AttrValue attr_tensor; in AddNodeConstScalar() local446 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodeConstScalar()447 node->mutable_attr()->insert({"value", attr_tensor}); in AddNodeConstScalar()1350 AttrValue attr_tensor; in AddNodeShapeConst() local1356 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodeShapeConst()1357 node->mutable_attr()->insert({"value", attr_tensor}); in AddNodeShapeConst()
974 AttrValue* attr_tensor) { in CreateConstantTensorAttrValue() argument975 TensorProto* t = attr_tensor->mutable_tensor(); in CreateConstantTensorAttrValue()1076 AttrValue attr_tensor; in CreateNodeDef() local1077 TensorProto* t = attr_tensor.mutable_tensor(); in CreateNodeDef()1145 node->mutable_attr()->insert({"value", attr_tensor}); in CreateNodeDef()