Searched refs:attr_output_shape (Results 1 – 2 of 2) sorted by relevance
642 AttrValue attr_output_shape; in UpdateFunction() local652 *attr_output_shape.mutable_shape() = proto; in UpdateFunction()653 (*fun_node->mutable_attr())["shape"] = attr_output_shape; in UpdateFunction()2268 AttrValue attr_output_shape; in AnnotateOutputShapes() local2271 *attr_output_shape.mutable_list()->add_shape() = tensor_property.shape(); in AnnotateOutputShapes()2273 (*node->mutable_attr())["_output_shapes"] = attr_output_shape; in AnnotateOutputShapes()
862 AttrValue attr_output_shape; in AddNodeTranspose() local863 auto output_shape = attr_output_shape.mutable_list()->add_shape(); in AddNodeTranspose()875 node->mutable_attr()->insert({"_output_shapes", attr_output_shape}); in AddNodeTranspose()