Searched refs:attr_tensor (Results 1 – 4 of 4) sorted by relevance
44 AttrValue attr_tensor; in AddNodeDivConst() local45 auto tensor = attr_tensor.mutable_tensor(); in AddNodeDivConst()48 node->mutable_attr()->insert({"value", attr_tensor}); in AddNodeDivConst()
425 AttrValue attr_tensor; in AddNodePermConst() local430 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodePermConst()431 node->mutable_attr()->insert({"value", attr_tensor}); in AddNodePermConst()451 AttrValue attr_tensor; in AddNodeConstScalar() local454 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodeConstScalar()455 node->mutable_attr()->insert({"value", attr_tensor}); in AddNodeConstScalar()1409 AttrValue attr_tensor; in AddNodeShapeConst() local1415 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodeShapeConst()1416 node->mutable_attr()->insert({"value", attr_tensor}); in AddNodeShapeConst()
310 AttrValue attr_tensor; in CreateConstPermNode() local315 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in CreateConstPermNode()316 node.mutable_attr()->insert({"value", attr_tensor}); in CreateConstPermNode()1257 AttrValue attr_tensor; in AddNodeShapeConst() local1264 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodeShapeConst()1265 new_node.mutable_attr()->insert({"value", attr_tensor}); in AddNodeShapeConst()
1131 AttrValue* attr_tensor) { in CreateConstantTensorAttrValue() argument1132 TensorProto* t = attr_tensor->mutable_tensor(); in CreateConstantTensorAttrValue()1236 AttrValue attr_tensor; in CreateNodeDef() local1237 TensorProto* t = attr_tensor.mutable_tensor(); in CreateNodeDef()1305 node->mutable_attr()->insert({"value", attr_tensor}); in CreateNodeDef()