Function generateFunctionsForOps

tensorflow/go/genop/internal/genop.go:107–130

(w io.Writer, ops *pb.OpList, apimap *apiDefMap)

}

func generateFunctionsForOps(w io.Writer, ops *pb.OpList, apimap *apiDefMap) error {
	thisPackage := reflect.TypeOf(tmplArgs{}).PkgPath()
	if err := tmplHeader.Execute(w, thisPackage); err != nil {
		return err
	}
	blacklist := map[string]bool{
		"Const":           true,
		"PyFunc":          true,
		"PyFuncStateless": true,
	}
	for _, op := range ops.Op {
		if blacklist[op.Name] {
			continue
		}
		apidef, err := apimap.Get(op.Name)
		if err != nil {
			return err
		}
		if err := generateFunctionForOp(w, op, apidef); err != nil {
			return err
		}
	}
	return nil
}

func generateFunctionForOp(w io.Writer, op *pb.OpDef, apidef *pb.ApiDef) error {
	if strings.HasPrefix(op.Name, "_") { // Internal operation