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()
307 AttrValue attr_tensor; in CreateConstPermNode() local312 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in CreateConstPermNode()313 node.mutable_attr()->insert({"value", attr_tensor}); in CreateConstPermNode()1252 AttrValue attr_tensor; in AddNodeShapeConst() local1259 tensor.AsProtoTensorContent(attr_tensor.mutable_tensor()); in AddNodeShapeConst()1260 new_node.mutable_attr()->insert({"value", attr_tensor}); in AddNodeShapeConst()
1140 AttrValue* attr_tensor) { in CreateConstantTensorAttrValue() argument1141 TensorProto* t = attr_tensor->mutable_tensor(); in CreateConstantTensorAttrValue()1247 AttrValue attr_tensor; in CreateNodeDef() local1248 TensorProto* t = attr_tensor.mutable_tensor(); in CreateNodeDef()1316 node->mutable_attr()->insert({"value", attr_tensor}); in CreateNodeDef()
782 attr_tensor = getattr(fx_graph_module, node.target)783 assert isinstance(attr_tensor, torch.Tensor), f"{attr_tensor} is not a tensor."789 value=attr_tensor,