Searched refs:shape_attr (Results 1 – 3 of 3) sorted by relevance
293 TensorShape shape_attr; in __anon3500048a0202() local294 TF_RETURN_IF_ERROR(c->GetAttr("shape", &shape_attr)); in __anon3500048a0202()296 TF_RETURN_IF_ERROR(c->MakeShapeFromTensorShape(shape_attr, &s)); in __anon3500048a0202()
145 tensorflow::AttrValue shape_attr; in TestPreserveOutputShapes() local146 auto* shape_proto = shape_attr.mutable_list()->add_shape(); in TestPreserveOutputShapes()153 placeholder.node()->AddAttr("_output_shapes", shape_attr); in TestPreserveOutputShapes()
1978 AttrValue shape_attr; in CreateIdentityGraphDef() local1979 shape_attr.mutable_shape()->set_unknown_rank(true); in CreateIdentityGraphDef()1985 placeholder->mutable_attr()->insert({"shape", shape_attr}); in CreateIdentityGraphDef()