Lines Matching defs:attr_name
650 string attr_name;
651 OP_REQUIRES_OK(context, context->GetAttr("attr_name", &attr_name));
653 status.emplace_back("s", context->GetAttr(attr_name, &s));
654 status.emplace_back("s_list", context->GetAttr(attr_name, &s_list));
655 status.emplace_back("i", context->GetAttr(attr_name, &i));
656 status.emplace_back("i_list", context->GetAttr(attr_name, &i_list));
657 status.emplace_back("i32", context->GetAttr(attr_name, &i32));
658 status.emplace_back("i32_list", context->GetAttr(attr_name, &i32_list));
659 status.emplace_back("f", context->GetAttr(attr_name, &f));
660 status.emplace_back("f_list", context->GetAttr(attr_name, &f_list));
661 status.emplace_back("b", context->GetAttr(attr_name, &b));
662 status.emplace_back("b_list", context->GetAttr(attr_name, &b_list));
663 status.emplace_back("type", context->GetAttr(attr_name, &type));
664 status.emplace_back("type_list", context->GetAttr(attr_name, &type_list));
666 context->GetAttr(attr_name, &type_vector));
668 context->GetAttr(attr_name, &shape_proto));
670 context->GetAttr(attr_name, &shape_proto_list));
671 status.emplace_back("shape", context->GetAttr(attr_name, &shape));
672 status.emplace_back("shape_list", context->GetAttr(attr_name, &shape_list));
713 .Attr("attr_name: string")
721 {"attr_name|string|'a'", "a|list(string)|['foo', 'bar']"});
727 {"attr_name|string|'b'", "a|list(string)|['baz']"});
734 .Attr("attr_name: string")
742 {"attr_name|string|'a'", "a|int|35", "b|list(int)|[-1, 2, -4]"});
750 {"attr_name|string|'b'", "a|int|35", "b|list(int)|[-1, 2, -4]"});
758 {"attr_name|string|'a'", "a|int|8589934592",
772 {"attr_name|string|'b'", "a|int|8589934592",
787 .Attr("attr_name: string")
795 {"attr_name|string|'a'", "a|shape|{ dim { size: 3 } }",
804 {"attr_name|string|'b'", "a|shape|{ dim { size: 3 } }",
818 REGISTER_OP("GetAttrType").Attr("attr_name: string").Attr("a: type");
823 "GetAttrType", DEVICE_CPU, {"attr_name|string|'a'", "a|type|DT_FLOAT"});
829 REGISTER_OP("GetAttrTypeList").Attr("attr_name: string").Attr("a: list(type)");
836 {"attr_name|string|'a'", "a|list(type)|[DT_INT32, DT_BOOL]"});
